Apriori_Trie.hpp

Go to the documentation of this file.

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

#ifndef Apriori_Trie_HPP
#define Apriori_Trie_HPP

#include "Trie.hpp"
#include "Input_Output_Manager.hpp"
#include <set>
#include <vector>
using namespace std;


class Apriori_Trie
{
public:

      Apriori_Trie( const unsigned long counter_of_emptyset );
   
      void insert_frequent_items(const vector<unsigned long>& counters );
   
      void candidate_generation( const itemtype& frequent_size,
                              Input_Output_Manager& input_output_manager );

      void find_candidate( const vector<itemtype>& basket, 
                           const itemtype candidate_size, 
                           const unsigned long counter=1 );

      void delete_infrequent( const unsigned long min_occurrence, 
                              const itemtype candidate_size );
      bool is_there_any_candidate() const;
      void statistics() const;
      
      ~Apriori_Trie();

protected:

      bool is_all_subset_frequent( const set<itemtype>& maybe_candidate ) const;

      void candidate_generation_two();

      void candidate_generation_assist( Trie* Trie, 
                                        const itemtype distance_from_generator,
                                        set<itemtype>& maybe_candidate,
                                        Input_Output_Manager& input_output_manager );

      void find_candidate_two( const vector<itemtype>& basket, 
                               const unsigned long counter=1 );

      void delete_infrequent_two( const unsigned long min_occurrence );

private:
   // No private methods

public:
   // No public members

protected:
      Trie main_trie;

      vector< vector<unsigned long> > temp_counter_array;
      vector<unsigned long> nr_of_freq_itemsets;
};

#endif

---