DecisionStump.h

Go to the documentation of this file.

#ifndef GRT_DECISION_STUMP_HEADER
#define GRT_DECISION_STUMP_HEADER

#include "WeakClassifier.h"

GRT_BEGIN_NAMESPACE
    
class GRT_API DecisionStump : public WeakClassifier{
public:
    DecisionStump(const UINT numRandomSplits = 100);
    
    virtual ~DecisionStump();
    
    DecisionStump(const DecisionStump &rhs);
    
    DecisionStump& operator=(const DecisionStump &rhs);
    
    virtual bool deepCopyFrom(const WeakClassifier *weakClassifer);
    
    virtual bool train(ClassificationData &trainingData, VectorFloat &weights);
    
    virtual Float predict(const VectorFloat &x);
    
    virtual bool saveModelToFile( std::fstream &file ) const;
    
    virtual bool loadModelFromFile( std::fstream &file );
    
    virtual void print() const;
    
    UINT getDecisionFeatureIndex() const;
    
    UINT getDirection() const;
    
    UINT getNumRandomSplits() const;
    
    Float getDecisionValue() const;

protected:
    UINT decisionFeatureIndex;  
    UINT direction;             
    UINT numRandomSplits;       
    Float decisionValue;       
    
    static RegisterWeakClassifierModule< DecisionStump > registerModule; 
};

GRT_END_NAMESPACE

#endif //GRT_DECISION_STUMP_HEADER