31 #ifndef GRT_TREE_HEADER
32 #define GRT_TREE_HEADER
50 Tree(
const UINT numSplittingSteps=100,
const UINT minNumSamplesPerNode=5,
const UINT maxDepth=10,
const bool removeFeaturesAtEachSpilt =
false,
const UINT trainingMode = BEST_ITERATIVE_SPILT);
70 virtual bool print()
const;
79 virtual bool getModel(ostream &stream)
const;
197 UINT numSplittingSteps;
198 UINT minNumSamplesPerNode;
200 bool removeFeaturesAtEachSpilt;
204 enum TrainingMode{BEST_ITERATIVE_SPILT=0,BEST_RANDOM_SPLIT,NUM_TRAINING_MODES};
210 #endif //GRT_TREE_HEADER
const Node * getTree() const
UINT getPredictedNodeID() const
bool setTrainingMode(const UINT trainingMode)
UINT getMinNumSamplesPerNode() const
bool getRemoveFeaturesAtEachSpilt() const
Tree(const UINT numSplittingSteps=100, const UINT minNumSamplesPerNode=5, const UINT maxDepth=10, const bool removeFeaturesAtEachSpilt=false, const UINT trainingMode=BEST_ITERATIVE_SPILT)
virtual bool getModel(ostream &stream) const
UINT getTrainingMode() const
bool setNumSplittingSteps(const UINT numSplittingSteps)
This class contains the main Node base class.
UINT getNumSplittingSteps() const
virtual Node * deepCopyTree() const
bool setRemoveFeaturesAtEachSpilt(const bool removeFeaturesAtEachSpilt)
virtual bool print() const
bool setMaxDepth(const UINT maxDepth)
bool setMinNumSamplesPerNode(const UINT minNumSamplesPerNode)