SepPlanExplore.hh

Go to the documentation of this file.

#ifndef _SEPPLANEXPLORE_HH_
#define _SEPPLANEXPLORE_HH_

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

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

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

class SepPlanExplore: public Classifier {

public:

  // mode - re-build stump every step?  
  // re-build only on misclassifications? or rebuild every 'trainFreq' steps
  SepPlanExplore(int id, int modelType, int predType, int nModels, 
                 int trainMode, int trainFreq,
                 float featPct, float expPct, float treeThreshold, bool stoch,
                 float featRange, Random rng);

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

  ~SepPlanExplore();

  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 initModels();

  bool SPEDEBUG;

private:

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

  Random rng;

  Classifier* expModel;
  Classifier* planModel;

};


#endif