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 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 saveModelToFile( std::fstream &file ) const;
    
    virtual bool loadModelFromFile( 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::saveModelToFile;
    using MLBase::loadModelFromFile;
    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