AdaBoost.h

Go to the documentation of this file.

#ifndef GRT_ADABOOST_HEADER
#define GRT_ADABOOST_HEADER

#include "../../CoreModules/Classifier.h"
#include "AdaBoostClassModel.h"
#include "WeakClassifiers/DecisionStump.h"
#include "WeakClassifiers/RadialBasisFunction.h"

GRT_BEGIN_NAMESPACE

//typedef DecisionStump AdaBoostWeakClassifier;

class GRT_API AdaBoost : public Classifier
{
 public:
 AdaBoost(const WeakClassifier &weakClassifier = DecisionStump(),bool useScaling=false,bool useNullRejection=false,Float nullRejectionCoeff=10.0,UINT numBoostingIterations=20,UINT predictionMethod=MAX_VALUE);
 
 AdaBoost(const AdaBoost &rhs);
 
 virtual ~AdaBoost();
 
 AdaBoost &operator=(const AdaBoost &rhs);
 
 virtual bool deepCopyFrom(const Classifier *classifier);
 
 virtual bool train_(ClassificationData &trainingData);
 
 virtual bool predict_(VectorFloat &inputVector);
 
 virtual bool clear();
 
 virtual bool save( std::fstream &file ) const;
 
 virtual bool load( std::fstream &file );
 
 virtual bool recomputeNullRejectionThresholds();
 
 bool setNullRejectionCoeff(Float nullRejectionCoeff);
 
 bool setWeakClassifier(const WeakClassifier &weakClassifer);
 
 bool addWeakClassifier(const WeakClassifier &weakClassifer);
 
 bool clearWeakClassifiers();
 
 bool setNumBoostingIterations(UINT numBoostingIterations);
 
 bool setPredictionMethod(UINT predictionMethod);
 
 void printModel();
 
 Vector< AdaBoostClassModel > getModels() const { return models; }
 
 //Tell the compiler we are using the following functions from the MLBase class to stop hidden virtual function warnings
 using MLBase::save;
 using MLBase::load;
 using MLBase::train;
 using MLBase::train_;
 using MLBase::predict;
 using MLBase::predict_;
 
protected:
 bool loadLegacyModelFromFile( std::fstream &file );
 
 UINT numBoostingIterations;
 UINT predictionMethod;
 Vector< WeakClassifier* > weakClassifiers;
 Vector< AdaBoostClassModel > models;
 
 static RegisterClassifierModule< AdaBoost > registerModule;
 
public:
 enum PredictionMethods{MAX_POSITIVE_VALUE=0,MAX_VALUE};
};

GRT_END_NAMESPACE

#endif //GRT_ADABOOST_HEADER