00001 /*************************************************************************** 00002 apriori.h - description 00003 ------------------- 00004 begin : cs dec 26 2002 00005 copyright : (C) 2002 by Ferenc Bodon 00006 email : bodon@mit.bme.hu 00007 ***************************************************************************/ 00008 00009 #ifndef APRIORI_H 00010 #define APRIORI_H 00011 00012 #include "Apriori_Trie.hpp" 00013 #include <map> 00014 00015 00078 class Apriori { 00079 public: 00080 Apriori( ifstream& basket_file, const char* output_file_name, 00081 const bool store_input ); 00082 00084 void APRIORI_alg( const double min_supp, const double min_conf, 00085 const bool quiet, const unsigned long size_threshold ); 00086 ~Apriori(); 00087 private: 00088 00090 void support( const itemtype& candidate_size ); 00091 00092 protected: 00093 // No protected class data members 00094 00095 private: 00097 Apriori_Trie* apriori_trie; 00100 Input_Output_Manager input_output_manager; 00102 map<vector<itemtype>, unsigned long> reduced_baskets; 00105 bool store_input; 00106 }; 00107 00108 #endif