35 #ifndef GRT_DECISION_TREE_CLUSTER_NODE_HEADER
36 #define GRT_DECISION_TREE_CLUSTER_NODE_HEADER
39 #include "../../ClusteringModules/KMeans/KMeans.h"
65 virtual bool predict(
const VectorDouble &x);
81 virtual bool print()
const;
90 virtual bool getModel(ostream &stream)
const;
131 bool set(
const UINT nodeSize,
const UINT featureIndex,
const double threshold,
const VectorDouble &classProbabilities);
135 virtual bool computeBestSpiltBestIterativeSpilt(
const UINT &numSplittingSteps,
const ClassificationData &trainingData,
const vector< UINT > &features,
const vector< UINT > &classLabels, UINT &featureIndex,
double &minError );
137 virtual bool computeBestSpiltBestRandomSpilt(
const UINT &numSplittingSteps,
const ClassificationData &trainingData,
const vector< UINT > &features,
const vector< UINT > &classLabels, UINT &featureIndex,
double &minError );
139 bool computeBestSpilt(
const UINT &numSplittingSteps,
const ClassificationData &trainingData,
const vector< UINT > &features,
const vector< UINT > &classLabels, UINT &featureIndex,
double &minError );
166 #endif //GRT_DECISION_TREE_DOT_PRODUCT_NODE_HEADER
virtual bool print() const
virtual Node * deepCopyNode() const
UINT getFeatureIndex() const
virtual ~DecisionTreeClusterNode()
DecisionTreeClusterNode * deepCopy() const
double getThreshold() const
DecisionTreeClusterNode()
virtual bool getModel(ostream &stream) const
This file implements a DecisionTreeNode, which is a specific base node used for a DecisionTree...
bool set(const UINT nodeSize, const UINT featureIndex, const double threshold, const VectorDouble &classProbabilities)
virtual bool loadParametersFromFile(fstream &file)
virtual bool saveParametersToFile(fstream &file) const
virtual bool predict(const VectorDouble &x)