Tree.h

Go to the documentation of this file.

#ifndef GRT_TREE_HEADER
#define GRT_TREE_HEADER

#include "Node.h"

GRT_BEGIN_NAMESPACE

class GRT_API Tree : public GRTBase
{
public:
 enum TrainingMode{BEST_ITERATIVE_SPILT=0,BEST_RANDOM_SPLIT,NUM_TRAINING_MODES};
 
 Tree(const UINT numSplittingSteps=100,const UINT minNumSamplesPerNode=5,const UINT maxDepth=10,const bool removeFeaturesAtEachSpilt = false,const UINT trainingMode = BEST_ITERATIVE_SPILT);
 
 virtual ~Tree(void);
 
 virtual bool clear();
 
 virtual bool print() const;
 
 virtual bool getModel( std::ostream &stream ) const;
 
 virtual Node* deepCopyTree() const;
 
 const Node* getTree() const;
 
 UINT getTrainingMode() const;
 
 UINT getNumSplittingSteps() const;
 
 UINT getMinNumSamplesPerNode() const;
 
 UINT getMaxDepth() const;
 
 UINT getPredictedNodeID() const;
 
 bool getRemoveFeaturesAtEachSpilt() const;
 
 bool setTrainingMode(const UINT trainingMode);
 
 bool setNumSplittingSteps(const UINT numSplittingSteps);
 
 bool setMinNumSamplesPerNode(const UINT minNumSamplesPerNode);
 
 bool setMaxDepth(const UINT maxDepth);
 
 bool setRemoveFeaturesAtEachSpilt(const bool removeFeaturesAtEachSpilt);
 
protected:
 
 UINT trainingMode;
 UINT numSplittingSteps;
 UINT minNumSamplesPerNode;
 UINT maxDepth;
 bool removeFeaturesAtEachSpilt;
 Node *tree; 
};

GRT_END_NAMESPACE

#endif //GRT_TREE_HEADER