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 00069 class Apriori { 00070 public: 00071 Apriori( const bool& quiet, const bool& store_input ); 00072 00074 void APRIORI_alg( ofstream& outcomefile, const char* basket_filename, const double& min_supp, const double& min_conf ); 00075 00076 private: 00077 00079 void read_in_a_line( FILE* filepoint ); 00081 void support( FILE* filepoint, const itemtype& candidate_size ); 00082 00083 Apriori_Trie* apriori_trie; 00084 vector<itemtype> basket; 00085 unsigned long basket_number; 00086 map<vector<itemtype>, unsigned long> reduced_baskets; 00087 bool quiet; 00088 bool store_input; 00089 }; 00090 00091 #endif