MultipleClassifiers.hh

Go to the documentation of this file.

#ifndef _MULTCLASS_HH_
#define _MULTCLASS_HH_

#include "../Models/C45Tree.hh"
#include "../Models/M5Tree.hh"
#include "../Models/LinearSplitsTree.hh"

#include "../Models/Stump.hh"

#include <rl_common/Random.h>
#include <rl_common/core.hh>
#include <vector>
#include <set>
#include <map>

class MultipleClassifiers: public Classifier {

public:

  MultipleClassifiers(int id, int modelType, int predType, int nModels, 
                      int trainMode, int trainFreq,
                      float featPct, float expPct, float treeThreshold,
                      bool stoch, float featRange, Random rng);

  MultipleClassifiers(const MultipleClassifiers&);
  virtual MultipleClassifiers* getCopy();

  ~MultipleClassifiers();

  virtual bool trainInstances(std::vector<classPair> &instances);
  virtual bool trainInstance(classPair &instance);
  virtual void testInstance(const std::vector<float> &input, std::map<float, float>* retval);
  virtual float getConf(const std::vector<float> &s);
  
  void updateModelAccuracy(int i, const std::vector<float> &input, float out);

  void initModels();

  float klDivergence(const std::vector<float> &input);

  float variance(const std::vector<float> &input);

  bool STDEBUG;
  bool PRED_DEBUG;
  bool ACC_DEBUG;
  bool CONF_DEBUG;
  bool COPYDEBUG;

  std::vector<Classifier*> models;

private:

  const int id;
  const int modelType;
  const int predType;
  const int nModels;
  const int mode;
  const int freq;
  float featPct;
  float expPct;
  const float treeThresh;
  const bool stoch;
  const bool addNoise;
  const float featRange;
  
  Random rng;

  std::vector<float> accuracy;
  int nsteps;
  std::vector<std::map<float, float> >infos;

};


#endif