DecisionTreeClusterNode.h

Go to the documentation of this file.

#ifndef GRT_DECISION_TREE_CLUSTER_NODE_HEADER
#define GRT_DECISION_TREE_CLUSTER_NODE_HEADER

#include "DecisionTreeNode.h"
#include "../../ClusteringModules/KMeans/KMeans.h"

GRT_BEGIN_NAMESPACE
    
class DecisionTreeClusterNode : public DecisionTreeNode{
public:
    DecisionTreeClusterNode();
    
    virtual ~DecisionTreeClusterNode();
    
    virtual bool predict(const VectorFloat &x);
    
    virtual bool clear();
    
    virtual bool print() const;

    virtual bool computeFeatureWeights( VectorFloat &weights ) const;

    virtual bool computeLeafNodeWeights( MatrixFloat &weights ) const;
    
    virtual bool getModel( std::ostream &stream ) const;
    
    virtual Node* deepCopyNode() const;
    
    DecisionTreeClusterNode* deepCopy() const;
    
    UINT getFeatureIndex() const;
    
    Float getThreshold() const;
    
    bool set(const UINT nodeSize,const UINT featureIndex,const Float threshold,const VectorFloat &classProbabilities);
    
protected:
    
    virtual bool computeBestSpiltBestIterativeSpilt( const UINT &numSplittingSteps, const ClassificationData &trainingData, const Vector< UINT > &features, const Vector< UINT > &classLabels, UINT &featureIndex, Float &minError );
    
    virtual bool computeBestSpiltBestRandomSpilt( const UINT &numSplittingSteps, const ClassificationData &trainingData, const Vector< UINT > &features, const Vector< UINT > &classLabels, UINT &featureIndex, Float &minError );
    
    bool computeBestSpilt( const UINT &numSplittingSteps, const ClassificationData &trainingData, const Vector< UINT > &features, const Vector< UINT > &classLabels, UINT &featureIndex, Float &minError );

    bool computeError( const ClassificationData &trainingData, MatrixFloat &data, const Vector< UINT > &classLabels, Vector< MinMax > ranges, Vector< UINT > groupIndex, const UINT featureIndex, Float &threshold, Float &error );

    virtual bool saveParametersToFile( std::fstream &file ) const;
    
    virtual bool loadParametersFromFile( std::fstream &file );
    
    UINT featureIndex;
    Float threshold;
    
    static RegisterNode< DecisionTreeClusterNode > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_DECISION_TREE_CLUSTER_NODE_HEADER