Tree.h

Go to the documentation of this file.

#ifndef GRT_TREE_HEADER
#define GRT_TREE_HEADER

#include "Node.h"

GRT_BEGIN_NAMESPACE

class Tree : public GRTBase
{
public:
 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;
    
public:
    enum TrainingMode{BEST_ITERATIVE_SPILT=0,BEST_RANDOM_SPLIT,NUM_TRAINING_MODES};
    
};

GRT_END_NAMESPACE

#endif //GRT_TREE_HEADER