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