Apriori.hpp

Go to the documentation of this file.

/***************************************************************************
                          apriori.h  -  description
                             -------------------
    begin                : cs dec 26 2002
    copyright            : (C) 2002 by Ferenc Bodon
    email                : bodon@cs.bme.hu
 ***************************************************************************/

#ifndef APRIORI_H
#define APRIORI_H

#include "Apriori_Trie.hpp"
#include <map>


class Apriori {
public:

   Apriori( ifstream& basket_file, const char* output_file_name, 
      const bool store_input ):
      input_output_manager(basket_file, output_file_name ),
      store_input(store_input){}

   void APRIORI_alg( const double min_supp, 
                     const bool quiet, const countertype size_threshold );
    ~Apriori(){delete apriori_trie;}
private:

   void support( const itemtype& candidate_size );

protected:
   // No protected class data members

private:
   Apriori_Trie*                           apriori_trie;

   Input_Output_Manager                    input_output_manager;

   map<vector<itemtype>, countertype>    reduced_baskets;

   bool                                    store_input;
};

#endif

---