31 #ifndef GRT_REGRESSION_TREE_HEADER
32 #define GRT_REGRESSION_TREE_HEADER
34 #include "../../CoreModules/Regressifier.h"
35 #include "../../CoreAlgorithms/Tree/Tree.h"
54 RegressionTree(
const UINT numSplittingSteps=100,
const UINT minNumSamplesPerNode=5,
const UINT maxDepth=10,
const bool removeFeaturesAtEachSpilt =
false,
const UINT trainingMode = BEST_ITERATIVE_SPILT,
const bool useScaling=
false,
const double minRMSErrorPerNode = 0.01);
101 virtual bool predict_(VectorDouble &inputVector);
109 virtual bool clear();
116 virtual bool print()
const;
177 bool computeBestSpilt(
const RegressionData &trainingData,
const vector< UINT > &features, UINT &featureIndex,
double &threshold,
double &minError );
178 bool computeBestSpiltBestIterativeSpilt(
const RegressionData &trainingData,
const vector< UINT > &features, UINT &featureIndex,
double &threshold,
double &minError );
180 bool computeNodeRegressionData(
const RegressionData &trainingData, VectorDouble ®ressionData );
188 #endif //GRT_REGRESSION_TREE_HEADER
virtual bool saveModelToFile(string filename) const
virtual bool loadModelFromFile(fstream &file)
RegressionTree & operator=(const RegressionTree &rhs)
virtual bool loadModelFromFile(string filename)
double minRMSErrorPerNode
double getMinRMSErrorPerNode() const
RegressionTree(const UINT numSplittingSteps=100, const UINT minNumSamplesPerNode=5, const UINT maxDepth=10, const bool removeFeaturesAtEachSpilt=false, const UINT trainingMode=BEST_ITERATIVE_SPILT, const bool useScaling=false, const double minRMSErrorPerNode=0.01)
virtual ~RegressionTree(void)
virtual bool train(ClassificationData trainingData)
This file implements a RegressionTreeNode, which is a specific type of node used for a RegressionTree...
const RegressionTreeNode * getTree() const
virtual bool saveModelToFile(fstream &file) const
virtual bool train_(RegressionData &trainingData)
virtual bool predict_(VectorDouble &inputVector)
virtual bool predict(VectorDouble inputVector)
virtual bool deepCopyFrom(const Regressifier *regressifier)
bool setMinRMSErrorPerNode(const double minRMSErrorPerNode)
virtual bool print() const
RegressionTreeNode * deepCopyTree() const