UnigramTextClassifier.h

Go to the documentation of this file.

#ifndef _UnigramTextClassifier_H_
#define _UnigramTextClassifier_H_
#include <map>
#include <iostream>
#include <fstream>

namespace std 
{
  typedef map<unsigned char,unsigned long> frequency_map;

  class UnigramTextClassifier 
  {
  public:
   
    UnigramTextClassifier();

    UnigramTextClassifier(const string classification);

    frequency_map freqs() { return _freqs; }
    unsigned long corpus_total() { return _corpus_total; }
    unsigned long total() { return _total; }
    string classification() { return _classification; }
    void setClassification(string& classification) {_classification = classification;}

    void UnigramTextClassifier::learn(istream& in);

    void UnigramTextClassifier::learn(char* in);

    void UnigramTextClassifier::dump(ostream& out);
 
    void UnigramTextClassifier::dump(char* out);

    void UnigramTextClassifier::read(istream& in);

    void UnigramTextClassifier::read(char* in);
 
    float UnigramTextClassifier::score(istream& in); 
//    {return ( (total()*8)-(bits_required(in)) ); }
    float UnigramTextClassifier::score(char* in) ;
//    {return ( (total()*8)-(bits_required(in)) ); }

    float UnigramTextClassifier::bits_required(unsigned char ch);

    float UnigramTextClassifier::bits_required(istream& in);
   
    float UnigramTextClassifier::bits_required(char* in);
  private:
    frequency_map _freqs;
    unsigned long _corpus_total;
    unsigned long _total;
    string _classification;
    float UnigramTextClassifier::lg (float n);
    float UnigramTextClassifier::info_value(float n);
    string UnigramTextClassifier::ctime_string();
  };

}
using namespace std;

#endif /* _UnigramTextClassifier_H_ */

---