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 MLBase
{
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 TrainingMode trainingMode = BEST_ITERATIVE_SPILT);
    
    virtual ~Tree(void);
    
    virtual bool clear() override;
    
    virtual bool print() const override;
    
    virtual bool getModel( std::ostream &stream ) const override;
    
    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 TrainingMode trainingMode);
    
    bool setNumSplittingSteps(const UINT numSplittingSteps);
    
    bool setMinNumSamplesPerNode(const UINT minNumSamplesPerNode);
    
    bool setMaxDepth(const UINT maxDepth);
    
    bool setRemoveFeaturesAtEachSpilt(const bool removeFeaturesAtEachSpilt);
    
protected:
    
    TrainingMode trainingMode;
    UINT numSplittingSteps;
    UINT minNumSamplesPerNode;
    UINT maxDepth;
    bool removeFeaturesAtEachSpilt;
    Node *tree; 
};

GRT_END_NAMESPACE

#endif //GRT_TREE_HEADER