DecisionTreeThresholdNode.h

Go to the documentation of this file.

#ifndef GRT_DECISION_TREE_THRESHOLD_NODE_HEADER
#define GRT_DECISION_TREE_THRESHOLD_NODE_HEADER

#include "DecisionTreeNode.h"

GRT_BEGIN_NAMESPACE
 
class GRT_API DecisionTreeThresholdNode : public DecisionTreeNode{
public:
 DecisionTreeThresholdNode();
 
 virtual ~DecisionTreeThresholdNode();
 
 virtual bool predict(const VectorFloat &x);
 
 virtual bool clear();
 
 virtual bool print() const;
 
 virtual bool getModel( std::ostream &stream ) const;
 
 virtual Node* deepCopyNode() const;
 
 DecisionTreeThresholdNode* deepCopy() const;
 
 UINT getFeatureIndex() const;
 Float getThreshold() const;
 
 bool set(const UINT nodeSize,const UINT featureIndex,const Float threshold,const VectorFloat &classProbabilities);
 
protected:
 
 virtual bool computeBestSpiltBestIterativeSpilt( const UINT &numSplittingSteps, const ClassificationData &trainingData, const Vector< UINT > &features, const Vector< UINT > &classLabels, UINT &featureIndex, Float &minError );
 
 virtual bool computeBestSpiltBestRandomSpilt( const UINT &numSplittingSteps, const ClassificationData &trainingData, const Vector< UINT > &features, const Vector< UINT > &classLabels, UINT &featureIndex, Float &minError );
 
 virtual bool saveParametersToFile( std::fstream &file ) const;
 
 virtual bool loadParametersFromFile( std::fstream &file );
 
 UINT featureIndex;
 Float threshold;
 
 static RegisterNode< DecisionTreeThresholdNode > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_DECISION_TREE_THRESHOLD_NODE_HEADER