DecisionTree.cpp

/*
GRT MIT License
Copyright (c) <2012> <Nicholas Gillian, Media Lab, MIT>

Permission is hereby granted, free of charge, to any person obtaining a copy of this software 
and associated documentation files (the "Software"), to deal in the Software without restriction, 
including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, 
and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, 
subject to the following conditions:

The above copyright notice and this permission notice shall be included in all copies or substantial 
portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT 
LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. 
IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, 
WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE 
SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/

#include "DecisionTree.h"

GRT_BEGIN_NAMESPACE

//Register the DecisionTree module with the Classifier base class
RegisterClassifierModule< DecisionTree > DecisionTree::registerModule("DecisionTree");

DecisionTree::DecisionTree(const DecisionTreeNode &decisionTreeNode,const UINT minNumSamplesPerNode,const UINT maxDepth,const bool removeFeaturesAtEachSpilt,const UINT trainingMode,const UINT numSplittingSteps,const bool useScaling)
{
 this->tree = NULL;
 this->decisionTreeNode = NULL;
 this->minNumSamplesPerNode = minNumSamplesPerNode;
 this->maxDepth = maxDepth;
 this->removeFeaturesAtEachSpilt = removeFeaturesAtEachSpilt;
 this->trainingMode = trainingMode;
 this->numSplittingSteps = numSplittingSteps;
 this->useScaling = useScaling;
 this->supportsNullRejection = true;
 Classifier::classType = "DecisionTree";
 classifierType = Classifier::classType;
 classifierMode = STANDARD_CLASSIFIER_MODE;
 Classifier::debugLog.setProceedingText("[DEBUG DecisionTree]");
 Classifier::errorLog.setProceedingText("[ERROR DecisionTree]");
 Classifier::trainingLog.setProceedingText("[TRAINING DecisionTree]");
 Classifier::warningLog.setProceedingText("[WARNING DecisionTree]");
 
 this->decisionTreeNode = decisionTreeNode.deepCopy();
 
}
 
DecisionTree::DecisionTree(const DecisionTree &rhs){
 tree = NULL;
 decisionTreeNode = NULL;
 Classifier::classType = "DecisionTree";
 classifierType = Classifier::classType;
 classifierMode = STANDARD_CLASSIFIER_MODE;
 Classifier:: debugLog.setProceedingText("[DEBUG DecisionTree]");
 Classifier::errorLog.setProceedingText("[ERROR DecisionTree]");
 Classifier::trainingLog.setProceedingText("[TRAINING DecisionTree]");
 Classifier::warningLog.setProceedingText("[WARNING DecisionTree]");
 *this = rhs;
}

DecisionTree::~DecisionTree(void)
{
 clear();
 
 if( decisionTreeNode != NULL ){
 delete decisionTreeNode;
 decisionTreeNode = NULL;
 }
}
 
DecisionTree& DecisionTree::operator=(const DecisionTree &rhs){
 if( this != &rhs ){
 //Clear this tree
 clear();
 
 if( rhs.getTrained() ){
 //Deep copy the tree
 this->tree = (DecisionTreeNode*)rhs.deepCopyTree();
 }
 
 //Deep copy the main node
 if( this->decisionTreeNode != NULL ){
 delete decisionTreeNode;
 decisionTreeNode = NULL;
 }
 this->decisionTreeNode = rhs.deepCopyDecisionTreeNode();
 
 this->minNumSamplesPerNode = rhs.minNumSamplesPerNode;
 this->maxDepth = rhs.maxDepth;
 this->removeFeaturesAtEachSpilt = rhs.removeFeaturesAtEachSpilt;
 this->trainingMode = rhs.trainingMode;
 this->numSplittingSteps = rhs.numSplittingSteps;
 this->nodeClusters = rhs.nodeClusters;

 //Copy the base classifier variables
 copyBaseVariables( (Classifier*)&rhs );
 }
 return *this;
}

bool DecisionTree::deepCopyFrom(const Classifier *classifier){
 
 if( classifier == NULL ) return false;
 
 if( this->getClassifierType() == classifier->getClassifierType() ){
 
 DecisionTree *ptr = (DecisionTree*)classifier;
 
 //Clear this tree
 this->clear();
 
 if( ptr->getTrained() ){
 //Deep copy the tree
 this->tree = ptr->deepCopyTree();
 }
 
 //Deep copy the main node
 if( this->decisionTreeNode != NULL ){
 delete decisionTreeNode;
 decisionTreeNode = NULL;
 }
 this->decisionTreeNode = ptr->deepCopyDecisionTreeNode();
 
 this->minNumSamplesPerNode = ptr->minNumSamplesPerNode;
 this->maxDepth = ptr->maxDepth;
 this->removeFeaturesAtEachSpilt = ptr->removeFeaturesAtEachSpilt;
 this->trainingMode = ptr->trainingMode;
 this->numSplittingSteps = ptr->numSplittingSteps;
 this->nodeClusters = ptr->nodeClusters;
 
 //Copy the base classifier variables
 return copyBaseVariables( classifier );
 }
 return false;
}

bool DecisionTree::train_(ClassificationData &trainingData){
 
 //Clear any previous model
 clear();
 
 if( decisionTreeNode == NULL ){
 Classifier::errorLog << "train_(ClassificationData &trainingData) - The decision tree node has not been set! You must set this first before training a model." << std::endl;
 return false;
 }
 
 const unsigned int M = trainingData.getNumSamples();
 const unsigned int N = trainingData.getNumDimensions();
 const unsigned int K = trainingData.getNumClasses();
 
 if( M == 0 ){
 Classifier::errorLog << "train_(ClassificationData &trainingData) - Training data has zero samples!" << std::endl;
 return false;
 }
 
 numInputDimensions = N;
 numClasses = K;
 classLabels = trainingData.getClassLabels();
 ranges = trainingData.getRanges();

 //Get the validation set if needed
 ClassificationData validationData;
 if( useValidationSet ){
 validationData = trainingData.partition( validationSetSize );
 validationSetAccuracy = 0;
 validationSetPrecision.resize( useNullRejection ? K+1 : K, 0 );
 validationSetRecall.resize( useNullRejection ? K+1 : K, 0 );
 }
 
 //Scale the training data if needed
 if( useScaling ){
 //Scale the training data between 0 and 1
 trainingData.scale(0, 1);
 }
 
 //If we are using null rejection, then we need a copy of the training dataset for later
 ClassificationData trainingDataCopy;
 if( useNullRejection ){
 trainingDataCopy = trainingData;
 }
 
 //Setup the valid features - at this point all features can be used
 Vector< UINT > features(N);
 for(UINT i=0; i<N; i++){
 features[i] = i;
 }
 
 //Build the tree
 UINT nodeID = 0;
 tree = buildTree( trainingData, NULL, features, classLabels, nodeID );
 
 if( tree == NULL ){
 clear();
 Classifier::errorLog << "train_(ClassificationData &trainingData) - Failed to build tree!" << std::endl;
 return false;
 }
 
 //Flag that the algorithm has been trained
 trained = true;
 
 //Compute the null rejection thresholds if null rejection is enabled
 if( useNullRejection ){
 VectorFloat classLikelihoods( numClasses );
 Vector< UINT > predictions(M);
 VectorFloat distances(M);
 VectorFloat classCounter( numClasses, 0 );
 
 //Run over the training dataset and compute the distance between each training sample and the predicted node cluster
 for(UINT i=0; i<M; i++){
 //Run the prediction for this sample
 if( !tree->predict( trainingDataCopy[i].getSample(), classLikelihoods ) ){
 Classifier::errorLog << "predict_(VectorFloat &inputVector) - Failed to predict!" << std::endl;
 return false;
 }
 
 //Store the predicted class index and cluster distance
 predictions[i] = Util::getMaxIndex( classLikelihoods );
 distances[i] = getNodeDistance(trainingDataCopy[i].getSample(), tree->getPredictedNodeID() );
 
 classCounter[ predictions[i] ]++;
 }
 
 //Compute the average distance for each class between the training data and the node clusters
 classClusterMean.clear();
 classClusterStdDev.clear();
 classClusterMean.resize( numClasses, 0 );
 classClusterStdDev.resize( numClasses, 0.01 ); //we start the std dev with a small value to ensure it is not zero
 
 for(UINT i=0; i<M; i++){
 classClusterMean[ predictions[i] ] += distances[ i ];
 }
 for(UINT k=0; k<numClasses; k++){
 classClusterMean[k] /= MAX( classCounter[k], 1 );
 }
 
 //Compute the std deviation
 for(UINT i=0; i<M; i++){
 classClusterStdDev[ predictions[i] ] += MLBase::SQR( distances[ i ] - classClusterMean[ predictions[i] ] );
 }
 for(UINT k=0; k<numClasses; k++){
 classClusterStdDev[k] = sqrt( classClusterStdDev[k] / MAX( classCounter[k], 1 ) );
 }
 
 //Compute the null rejection thresholds using the class mean and std dev
 recomputeNullRejectionThresholds();
 }

 if( useValidationSet ){
 const UINT numTestSamples = validationData.getNumSamples();
 double numCorrect = 0;
 UINT testLabel = 0;
 VectorDouble testSample;
 VectorDouble validationSetPrecisionCounter( validationSetPrecision.size(), 0.0 );
 VectorDouble validationSetRecallCounter( validationSetRecall.size(), 0.0 );
 Classifier::trainingLog << "Testing model with validation set..." << std::endl;
 for(UINT i=0; i<numTestSamples; i++){
 testLabel = validationData[i].getClassLabel();
 testSample = validationData[i].getSample();
 predict_( testSample );
 if( predictedClassLabel == testLabel ){
 numCorrect++;
 validationSetPrecision[ getClassLabelIndexValue( testLabel ) ]++;
 validationSetRecall[ getClassLabelIndexValue( testLabel ) ]++;
 } 
 validationSetPrecisionCounter[ getClassLabelIndexValue( predictedClassLabel ) ]++;
 validationSetRecallCounter[ getClassLabelIndexValue( testLabel ) ]++;
 }

 validationSetAccuracy = (numCorrect / numTestSamples) * 100.0;
 for(size_t i=0; i<validationSetPrecision.size(); i++){
 validationSetPrecision[i] /= validationSetPrecisionCounter[i] > 0 ? validationSetPrecisionCounter[i] : 1;
 }
 for(size_t i=0; i<validationSetRecall.size(); i++){
 validationSetRecall[i] /= validationSetRecallCounter[i] > 0 ? validationSetRecallCounter[i] : 1;
 }

 Classifier::trainingLog << "Validation set accuracy: " << validationSetAccuracy << std::endl;

 Classifier::trainingLog << "Validation set precision: ";
 for(size_t i=0; i<validationSetPrecision.size(); i++){
 Classifier::trainingLog << validationSetPrecision[i] << " ";
 }
 Classifier::trainingLog << std::endl;

 Classifier::trainingLog << "Validation set recall: ";
 for(size_t i=0; i<validationSetRecall.size(); i++){
 Classifier::trainingLog << validationSetRecall[i] << " ";
 }
 Classifier::trainingLog << std::endl;
 }

 return true;
}

bool DecisionTree::predict_(VectorFloat &inputVector){
 
 predictedClassLabel = 0;
 maxLikelihood = 0;
 
 //Validate the input is OK and the model is trained properly
 if( !trained ){
 Classifier::errorLog << "predict_(VectorFloat &inputVector) - Model Not Trained!" << std::endl;
 return false;
 }
 
 if( tree == NULL ){
 Classifier::errorLog << "predict_(VectorFloat &inputVector) - DecisionTree pointer is null!" << std::endl;
 return false;
 }
 
 if( inputVector.getSize() != numInputDimensions ){
 Classifier::errorLog << "predict_(VectorFloat &inputVector) - The size of the input Vector (" << inputVector.getSize() << ") does not match the num features in the model (" << numInputDimensions << std::endl;
 return false;
 }
 
 //Scale the input data if needed
 if( useScaling ){
 for(UINT n=0; n<numInputDimensions; n++){
 inputVector[n] = grt_scale(inputVector[n], ranges[n].minValue, ranges[n].maxValue, 0.0, 1.0);
 }
 }
 
 if( classLikelihoods.size() != numClasses ) classLikelihoods.resize(numClasses,0);
 if( classDistances.size() != numClasses ) classDistances.resize(numClasses,0);
 
 //Run the decision tree prediction
 if( !tree->predict( inputVector, classLikelihoods ) ){
 Classifier::errorLog << "predict_(VectorFloat &inputVector) - Failed to predict!" << std::endl;
 return false;
 }
 
 //Find the maximum likelihood
 //The tree automatically returns proper class likelihoods so we don't need to do anything else
 UINT maxIndex = 0;
 maxLikelihood = 0;
 for(UINT k=0; k<numClasses; k++){
 if( classLikelihoods[k] > maxLikelihood ){
 maxLikelihood = classLikelihoods[k];
 maxIndex = k;
 }
 }
 
 //Run the null rejection
 if( useNullRejection ){
 
 //Get the distance between the input and the leaf mean
 Float leafDistance = getNodeDistance( inputVector, tree->getPredictedNodeID() );
 
 if( grt_isnan(leafDistance) ){
 Classifier::errorLog << "predict_(VectorFloat &inputVector) - Failed to match leaf node ID to compute node distance!" << std::endl;
 return false;
 }

 //Set the predicted class distance as the leaf distance, all other classes will have a distance of zero
 std::fill(classDistances.begin(),classDistances.end(),0);
 classDistances[ maxIndex ] = leafDistance;
 
 //Use the distance to check if the class label should be rejected or not
 if( leafDistance <= nullRejectionThresholds[ maxIndex ] ){
 predictedClassLabel = classLabels[ maxIndex ];
 }else predictedClassLabel = GRT_DEFAULT_NULL_CLASS_LABEL;
 
 }else {
 //Set the predicated class label
 predictedClassLabel = classLabels[ maxIndex ];
 }
 
 return true;
}
 
bool DecisionTree::clear(){
 
 //Clear the Classifier variables
 Classifier::clear();
 
 //Clear the node clusters
 nodeClusters.clear();
 
 //Delete the tree if it exists
 if( tree != NULL ){
 tree->clear();
 delete tree;
 tree = NULL;
 }
 
 //NOTE: We do not want to clean up the decisionTreeNode here as we need to keep track of this, this is only delete in the destructor
 
 return true;
}
 
bool DecisionTree::recomputeNullRejectionThresholds(){
 
 if( !trained ){
 Classifier::warningLog << "recomputeNullRejectionThresholds() - Failed to recompute null rejection thresholds, the model has not been trained!" << std::endl;
 return false;
 }

 if( !useNullRejection ){
 Classifier::warningLog << "recomputeNullRejectionThresholds() - Failed to recompute null rejection thresholds, null rejection is not enabled!" << std::endl;
 return false;
 }
 
 nullRejectionThresholds.resize( numClasses );
 
 //Compute the rejection threshold for each class using the mean and std dev
 for(UINT k=0; k<numClasses; k++){
 nullRejectionThresholds[k] = classClusterMean[k] + (classClusterStdDev[k]*nullRejectionCoeff);
 }
 
 return true;
}
 
bool DecisionTree::saveModelToFile( std::fstream &file ) const{
 
 if(!file.is_open())
 {
 Classifier::errorLog <<"saveModelToFile(fstream &file) - The file is not open!" << std::endl;
 return false;
 }
 
 //Write the header info
 file << "GRT_DECISION_TREE_MODEL_FILE_V4.0\n";
 
 //Write the classifier settings to the file
 if( !Classifier::saveBaseSettingsToFile(file) ){
 Classifier::errorLog <<"saveModelToFile(fstream &file) - Failed to save classifier base settings to file!" << std::endl;
 return false;
 }
 
 if( decisionTreeNode != NULL ){
 file << "DecisionTreeNodeType: " << decisionTreeNode->getNodeType() << std::endl;
 if( !decisionTreeNode->saveToFile( file ) ){
 Classifier::errorLog <<"saveModelToFile(fstream &file) - Failed to save decisionTreeNode settings to file!" << std::endl;
 return false;
 }
 }else{
 file << "DecisionTreeNodeType: " << "NULL" << std::endl;
 }
 
 file << "MinNumSamplesPerNode: " << minNumSamplesPerNode << std::endl;
 file << "MaxDepth: " << maxDepth << std::endl;
 file << "RemoveFeaturesAtEachSpilt: " << removeFeaturesAtEachSpilt << std::endl;
 file << "TrainingMode: " << trainingMode << std::endl;
 file << "NumSplittingSteps: " << numSplittingSteps << std::endl;
 file << "TreeBuilt: " << (tree != NULL ? 1 : 0) << std::endl;
 
 if( tree != NULL ){
 file << "Tree:\n";
 if( !tree->saveToFile( file ) ){
 Classifier::errorLog << "saveModelToFile(fstream &file) - Failed to save tree to file!" << std::endl;
 return false;
 }
 
 //Save the null rejection data if needed
 if( useNullRejection ){
 
 file << "ClassClusterMean:";
 for(UINT k=0; k<numClasses; k++){
 file << " " << classClusterMean[k];
 }
 file << std::endl;
 
 file << "ClassClusterStdDev:";
 for(UINT k=0; k<numClasses; k++){
 file << " " << classClusterStdDev[k];
 }
 file << std::endl;
 
 file << "NumNodes: " << nodeClusters.size() << std::endl;
 file << "NodeClusters:\n";
 
 std::map< UINT, VectorFloat >::const_iterator iter = nodeClusters.begin();
 
 while( iter != nodeClusters.end() ){
 
 //Write the nodeID
 file << iter->first;
 
 //Write the node cluster
 for(UINT j=0; j<numInputDimensions; j++){
 file << " " << iter->second[j];
 }
 file << std::endl;
 
 iter++;
 }
 }

 }
 
 return true;
}
 
bool DecisionTree::loadModelFromFile( std::fstream &file ){
 
 clear();
 
 if( decisionTreeNode != NULL ){
 delete decisionTreeNode;
 decisionTreeNode = NULL;
 }
 
 if( !file.is_open() )
 {
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not open file to load model" << std::endl;
 return false;
 }
 
 std::string word;
 file >> word;
 
 //Check to see if we should load a legacy file
 if( word == "GRT_DECISION_TREE_MODEL_FILE_V1.0" ){
 return loadLegacyModelFromFile_v1( file );
 }
 
 if( word == "GRT_DECISION_TREE_MODEL_FILE_V2.0" ){
 return loadLegacyModelFromFile_v2( file );
 }
 
 if( word == "GRT_DECISION_TREE_MODEL_FILE_V3.0" ){
 return loadLegacyModelFromFile_v3( file );
 }
 
 //Find the file type header
 if( word != "GRT_DECISION_TREE_MODEL_FILE_V4.0" ){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find Model File Header" << std::endl;
 return false;
 }
 
 //Load the base settings from the file
 if( !Classifier::loadBaseSettingsFromFile(file) ){
 Classifier::errorLog << "loadModelFromFile(string filename) - Failed to load base settings from file!" << std::endl;
 return false;
 }
 
 file >> word;
 if(word != "DecisionTreeNodeType:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the DecisionTreeNodeType!" << std::endl;
 return false;
 }
 file >> word;
 
 if( word != "NULL" ){
 
 decisionTreeNode = dynamic_cast< DecisionTreeNode* >( DecisionTreeNode::createInstanceFromString( word ) );

 if( decisionTreeNode == NULL ){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not create new DecisionTreeNode from type: " << word << std::endl;
 return false;
 }
 
 if( !decisionTreeNode->loadFromFile( file ) ){
 Classifier::errorLog <<"loadModelFromFile(fstream &file) - Failed to load decisionTreeNode settings from file!" << std::endl;
 return false;
 }
 }else{
 Classifier::errorLog <<"loadModelFromFile(fstream &file) - Failed to load decisionTreeNode! DecisionTreeNodeType is NULL!" << std::endl;
 return false;
 }
 
 file >> word;
 if(word != "MinNumSamplesPerNode:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the MinNumSamplesPerNode!" << std::endl;
 return false;
 }
 file >> minNumSamplesPerNode;
 
 file >> word;
 if(word != "MaxDepth:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the MaxDepth!" << std::endl;
 return false;
 }
 file >> maxDepth;
 
 file >> word;
 if(word != "RemoveFeaturesAtEachSpilt:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the RemoveFeaturesAtEachSpilt!" << std::endl;
 return false;
 }
 file >> removeFeaturesAtEachSpilt;
 
 file >> word;
 if(word != "TrainingMode:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the TrainingMode!" << std::endl;
 return false;
 }
 file >> trainingMode;
 
 file >> word;
 if(word != "NumSplittingSteps:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the NumSplittingSteps!" << std::endl;
 return false;
 }
 file >> numSplittingSteps;
 
 file >> word;
 if(word != "TreeBuilt:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the TreeBuilt!" << std::endl;
 return false;
 }
 file >> trained;
 
 if( trained ){
 file >> word;
 if(word != "Tree:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the Tree!" << std::endl;
 return false;
 }
 
 //Create a new DTree
 tree = dynamic_cast< DecisionTreeNode* >( decisionTreeNode->createNewInstance() );
 
 if( tree == NULL ){
 clear();
 Classifier::errorLog << "loadModelFromFile(fstream &file) - Failed to create new DecisionTreeNode!" << std::endl;
 return false;
 }
 
 tree->setParent( NULL );
 if( !tree->loadFromFile( file ) ){
 clear();
 Classifier::errorLog << "loadModelFromFile(fstream &file) - Failed to load tree from file!" << std::endl;
 return false;
 }
 
 //Load the null rejection data if needed
 if( useNullRejection ){
 
 UINT numNodes = 0;
 classClusterMean.resize( numClasses );
 classClusterStdDev.resize( numClasses );
 
 file >> word;
 if(word != "ClassClusterMean:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the ClassClusterMean header!" << std::endl;
 return false;
 }
 for(UINT k=0; k<numClasses; k++){
 file >> classClusterMean[k];
 }
 
 file >> word;
 if(word != "ClassClusterStdDev:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the ClassClusterStdDev header!" << std::endl;
 return false;
 }
 for(UINT k=0; k<numClasses; k++){
 file >> classClusterStdDev[k];
 }
 
 file >> word;
 if(word != "NumNodes:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the NumNodes header!" << std::endl;
 return false;
 }
 file >> numNodes;
 
 file >> word;
 if(word != "NodeClusters:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the NodeClusters header!" << std::endl;
 return false;
 }
 
 UINT nodeID = 0;
 VectorFloat cluster( numInputDimensions );
 for(UINT i=0; i<numNodes; i++){
 
 //load the nodeID
 file >> nodeID;
 
 for(UINT j=0; j<numInputDimensions; j++){
 file >> cluster[j];
 }
 
 //Add the cluster to the cluster nodes map
 nodeClusters[ nodeID ] = cluster;
 }
 
 //Recompute the null rejection thresholds
 recomputeNullRejectionThresholds();
 }
 
 //Resize the prediction results to make sure it is setup for realtime prediction
 maxLikelihood = DEFAULT_NULL_LIKELIHOOD_VALUE;
 bestDistance = DEFAULT_NULL_DISTANCE_VALUE;
 classLikelihoods.resize(numClasses,DEFAULT_NULL_LIKELIHOOD_VALUE);
 classDistances.resize(numClasses,DEFAULT_NULL_DISTANCE_VALUE);
 }
 
 return true;
}
 
bool DecisionTree::getModel( std::ostream &stream ) const{
 
 if( tree != NULL )
 return tree->getModel( stream );
 return false;

}
 
DecisionTreeNode* DecisionTree::deepCopyTree() const{
 
 if( tree == NULL ){
 return NULL;
 }
 
 return dynamic_cast< DecisionTreeNode* >( tree->deepCopyNode() );
}
 
DecisionTreeNode* DecisionTree::deepCopyDecisionTreeNode() const{
 
 if( decisionTreeNode == NULL ){
 return NULL;
 }
 
 return decisionTreeNode->deepCopy();
}
 
const DecisionTreeNode* DecisionTree::getTree() const{
 return dynamic_cast< DecisionTreeNode* >( tree );
}
 
bool DecisionTree::setDecisionTreeNode( const DecisionTreeNode &node ){
 
 if( decisionTreeNode != NULL ){
 delete decisionTreeNode;
 decisionTreeNode = NULL;
 }
 this->decisionTreeNode = node.deepCopy();
 
 return true;
}
 
DecisionTreeNode* DecisionTree::buildTree(ClassificationData &trainingData,DecisionTreeNode *parent,Vector< UINT > features,const Vector< UINT > &classLabels, UINT nodeID){
 
 const UINT M = trainingData.getNumSamples();
 const UINT N = trainingData.getNumDimensions();
 
 //Update the nodeID
 nodeID++;
 
 //Get the depth
 UINT depth = 0;
 
 if( parent != NULL )
 depth = parent->getDepth() + 1;
 
 //If there are no training data then return NULL
 if( trainingData.getNumSamples() == 0 )
 return NULL;
 
 //Create the new node
 DecisionTreeNode *node = dynamic_cast< DecisionTreeNode* >( decisionTreeNode->createNewInstance() );

 if( node == NULL )
 return NULL;

 //Get the class probabilities
 VectorFloat classProbs = trainingData.getClassProbabilities( classLabels ); 
 
 //Set the parent
 node->initNode( parent, depth, nodeID );
 
 //If all the training data belongs to the same class or there are no features left then create a leaf node and return
 if( trainingData.getNumClasses() == 1 || features.size() == 0 || M < minNumSamplesPerNode || depth >= maxDepth ){
 
 //Set the node
 node->setLeafNode( trainingData.getNumSamples(), classProbs );
 
 //Build the null cluster if null rejection is enabled
 if( useNullRejection ){
 nodeClusters[ nodeID ] = trainingData.getMean();
 }
 
 std::string info = "Reached leaf node.";
 if( trainingData.getNumClasses() == 1 ) info = "Reached pure leaf node.";
 else if( features.size() == 0 ) info = "Reached leaf node, no remaining features.";
 else if( M < minNumSamplesPerNode ) info = "Reached leaf node, hit min-samples-per-node limit.";
 else if( depth >= maxDepth ) info = "Reached leaf node, max depth reached.";
 
 Classifier::trainingLog << info << " Depth: " << depth << " NumSamples: " << trainingData.getNumSamples();

 Classifier::trainingLog << " Class Probabilities: ";
 for(UINT k=0; k<classProbs.getSize(); k++){
 Classifier::trainingLog << classProbs[k] << " ";
 }
 Classifier::trainingLog << std::endl;
 
 return node;
 }
 
 //Compute the best spilt point
 UINT featureIndex = 0;
 Float minError = 0;
 
 if( !node->computeBestSpilt( trainingMode, numSplittingSteps, trainingData, features, classLabels, featureIndex, minError ) ){
 delete node;
 return NULL;
 }
 
 Classifier::trainingLog << "Depth: " << depth << " FeatureIndex: " << featureIndex << " MinError: " << minError;
 Classifier::trainingLog << " Class Probabilities: ";
 for(size_t k=0; k<classProbs.size(); k++){
 Classifier::trainingLog << classProbs[k] << " ";
 }
 Classifier::trainingLog << std::endl;
 
 //Remove the selected feature so we will not use it again
 if( removeFeaturesAtEachSpilt ){
 for(size_t i=0; i<features.size(); i++){
 if( features[i] == featureIndex ){
 features.erase( features.begin()+i );
 break;
 }
 }
 }
 
 //Split the data into a left and right dataset
 ClassificationData lhs(N);
 ClassificationData rhs(N);
 
 //Reserve the memory to speed up the allocation of large datasets
 lhs.reserve( M );
 rhs.reserve( M );
 
 for(UINT i=0; i<M; i++){
 if( node->predict( trainingData[i].getSample() ) ){
 rhs.addSample(trainingData[i].getClassLabel(), trainingData[i].getSample());
 }else lhs.addSample(trainingData[i].getClassLabel(), trainingData[i].getSample());
 }
 
 //Clear the parent dataset so we do not run out of memory with very large datasets (with very deep trees)
 trainingData.clear();
 
 //Get the new node IDs for the children
 UINT leftNodeID = ++nodeID;
 UINT rightNodeID = ++nodeID;
 
 //Run the recursive tree building on the children
 node->setLeftChild( buildTree( lhs, node, features, classLabels, leftNodeID ) );
 node->setRightChild( buildTree( rhs, node, features, classLabels, rightNodeID ) );
 
 //Build the null clusters for the rhs and lhs nodes if null rejection is enabled
 if( useNullRejection ){
 nodeClusters[ leftNodeID ] = lhs.getMean();
 nodeClusters[ rightNodeID ] = rhs.getMean();
 }
 
 return node;
}
 
Float DecisionTree::getNodeDistance( const VectorFloat &x, const UINT nodeID ){
 
 //Use the node ID to find the node cluster
 std::map< UINT,VectorFloat >::iterator iter = nodeClusters.find( nodeID );
 
 //If we failed to find a match, return NAN
 if( iter == nodeClusters.end() ) return NAN;
 
 //Compute the distance between the input and the node cluster
 return getNodeDistance( x, iter->second );
}
 
Float DecisionTree::getNodeDistance( const VectorFloat &x, const VectorFloat &y ){
 
 Float distance = 0;
 const size_t N = x.size();
 
 for(size_t i=0; i<N; i++){
 distance += MLBase::SQR( x[i] - y[i] );
 }
 
 //Return the squared Euclidean distance instead of actual Euclidean distance as this is faster and just as useful
 return distance;
}
 
bool DecisionTree::loadLegacyModelFromFile_v1( std::fstream &file ){
 
 std::string word;
 
 file >> word;
 if(word != "NumFeatures:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find NumFeatures!" << std::endl;
 return false;
 }
 file >> numInputDimensions;
 
 file >> word;
 if(word != "NumClasses:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find NumClasses!" << std::endl;
 return false;
 }
 file >> numClasses;
 
 file >> word;
 if(word != "UseScaling:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find UseScaling!" << std::endl;
 return false;
 }
 file >> useScaling;
 
 file >> word;
 if(word != "UseNullRejection:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find UseNullRejection!" << std::endl;
 return false;
 }
 file >> useNullRejection;
 
 if( useScaling ){
 //Resize the ranges buffer
 ranges.resize( numInputDimensions );
 
 file >> word;
 if(word != "Ranges:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the Ranges!" << std::endl;
 return false;
 }
 for(UINT n=0; n<ranges.size(); n++){
 file >> ranges[n].minValue;
 file >> ranges[n].maxValue;
 }
 }
 
 file >> word;
 if(word != "NumSplittingSteps:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the NumSplittingSteps!" << std::endl;
 return false;
 }
 file >> numSplittingSteps;
 
 file >> word;
 if(word != "MinNumSamplesPerNode:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the MinNumSamplesPerNode!" << std::endl;
 return false;
 }
 file >> minNumSamplesPerNode;
 
 file >> word;
 if(word != "MaxDepth:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the MaxDepth!" << std::endl;
 return false;
 }
 file >> maxDepth;
 
 file >> word;
 if(word != "RemoveFeaturesAtEachSpilt:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the RemoveFeaturesAtEachSpilt!" << std::endl;
 return false;
 }
 file >> removeFeaturesAtEachSpilt;
 
 file >> word;
 if(word != "TrainingMode:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the TrainingMode!" << std::endl;
 return false;
 }
 file >> trainingMode;
 
 file >> word;
 if(word != "TreeBuilt:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the TreeBuilt!" << std::endl;
 return false;
 }
 file >> trained;
 
 if( trained ){
 file >> word;
 if(word != "Tree:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the Tree!" << std::endl;
 return false;
 }
 
 //Create a new DTree
 tree = new DecisionTreeNode;
 
 if( tree == NULL ){
 clear();
 Classifier::errorLog << "loadModelFromFile(fstream &file) - Failed to create new DecisionTreeNode!" << std::endl;
 return false;
 }
 
 tree->setParent( NULL );
 if( !tree->loadFromFile( file ) ){
 clear();
 Classifier::errorLog << "loadModelFromFile(fstream &file) - Failed to load tree from file!" << std::endl;
 return false;
 }
 }
 
 return true;
}
 
bool DecisionTree::loadLegacyModelFromFile_v2( std::fstream &file ){
 
 std::string word;
 
 //Load the base settings from the file
 if( !Classifier::loadBaseSettingsFromFile(file) ){
 Classifier::errorLog << "loadModelFromFile(string filename) - Failed to load base settings from file!" << std::endl;
 return false;
 }
 
 file >> word;
 if(word != "NumSplittingSteps:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the NumSplittingSteps!" << std::endl;
 return false;
 }
 file >> numSplittingSteps;
 
 file >> word;
 if(word != "MinNumSamplesPerNode:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the MinNumSamplesPerNode!" << std::endl;
 return false;
 }
 file >> minNumSamplesPerNode;
 
 file >> word;
 if(word != "MaxDepth:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the MaxDepth!" << std::endl;
 return false;
 }
 file >> maxDepth;
 
 file >> word;
 if(word != "RemoveFeaturesAtEachSpilt:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the RemoveFeaturesAtEachSpilt!" << std::endl;
 return false;
 }
 file >> removeFeaturesAtEachSpilt;
 
 file >> word;
 if(word != "TrainingMode:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the TrainingMode!" << std::endl;
 return false;
 }
 file >> trainingMode;
 
 file >> word;
 if(word != "TreeBuilt:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the TreeBuilt!" << std::endl;
 return false;
 }
 file >> trained;
 
 if( trained ){
 file >> word;
 if(word != "Tree:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the Tree!" << std::endl;
 return false;
 }
 
 //Create a new DTree
 tree = new DecisionTreeNode;
 
 if( tree == NULL ){
 clear();
 Classifier::errorLog << "loadModelFromFile(fstream &file) - Failed to create new DecisionTreeNode!" << std::endl;
 return false;
 }
 
 tree->setParent( NULL );
 if( !tree->loadFromFile( file ) ){
 clear();
 Classifier::errorLog << "loadModelFromFile(fstream &file) - Failed to load tree from file!" << std::endl;
 return false;
 }
 
 //Recompute the null rejection thresholds
 recomputeNullRejectionThresholds();
 
 //Resize the prediction results to make sure it is setup for realtime prediction
 maxLikelihood = DEFAULT_NULL_LIKELIHOOD_VALUE;
 bestDistance = DEFAULT_NULL_DISTANCE_VALUE;
 classLikelihoods.resize(numClasses,DEFAULT_NULL_LIKELIHOOD_VALUE);
 classDistances.resize(numClasses,DEFAULT_NULL_DISTANCE_VALUE);
 }
 
 return true;
}
 
bool DecisionTree::loadLegacyModelFromFile_v3( std::fstream &file ){
 
 std::string word;
 
 //Load the base settings from the file
 if( !Classifier::loadBaseSettingsFromFile(file) ){
 Classifier::errorLog << "loadModelFromFile(string filename) - Failed to load base settings from file!" << std::endl;
 return false;
 }
 
 file >> word;
 if(word != "NumSplittingSteps:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the NumSplittingSteps!" << std::endl;
 return false;
 }
 file >> numSplittingSteps;
 
 file >> word;
 if(word != "MinNumSamplesPerNode:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the MinNumSamplesPerNode!" << std::endl;
 return false;
 }
 file >> minNumSamplesPerNode;
 
 file >> word;
 if(word != "MaxDepth:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the MaxDepth!" << std::endl;
 return false;
 }
 file >> maxDepth;
 
 file >> word;
 if(word != "RemoveFeaturesAtEachSpilt:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the RemoveFeaturesAtEachSpilt!" << std::endl;
 return false;
 }
 file >> removeFeaturesAtEachSpilt;
 
 file >> word;
 if(word != "TrainingMode:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the TrainingMode!" << std::endl;
 return false;
 }
 file >> trainingMode;
 
 file >> word;
 if(word != "TreeBuilt:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the TreeBuilt!" << std::endl;
 return false;
 }
 file >> trained;
 
 if( trained ){
 file >> word;
 if(word != "Tree:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the Tree!" << std::endl;
 return false;
 }
 
 //Create a new DTree
 tree = new DecisionTreeNode;
 
 if( tree == NULL ){
 clear();
 Classifier::errorLog << "loadModelFromFile(fstream &file) - Failed to create new DecisionTreeNode!" << std::endl;
 return false;
 }
 
 tree->setParent( NULL );
 if( !tree->loadFromFile( file ) ){
 clear();
 Classifier::errorLog << "loadModelFromFile(fstream &file) - Failed to load tree from file!" << std::endl;
 return false;
 }
 
 //Load the null rejection data if needed
 if( useNullRejection ){
 
 UINT numNodes = 0;
 classClusterMean.resize( numClasses );
 classClusterStdDev.resize( numClasses );
 
 file >> word;
 if(word != "ClassClusterMean:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the ClassClusterMean header!" << std::endl;
 return false;
 }
 for(UINT k=0; k<numClasses; k++){
 file >> classClusterMean[k];
 }
 
 file >> word;
 if(word != "ClassClusterStdDev:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the ClassClusterStdDev header!" << std::endl;
 return false;
 }
 for(UINT k=0; k<numClasses; k++){
 file >> classClusterStdDev[k];
 }
 
 file >> word;
 if(word != "NumNodes:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the NumNodes header!" << std::endl;
 return false;
 }
 file >> numNodes;
 
 file >> word;
 if(word != "NodeClusters:"){
 Classifier::errorLog << "loadModelFromFile(string filename) - Could not find the NodeClusters header!" << std::endl;
 return false;
 }
 
 UINT nodeID = 0;
 VectorFloat cluster( numInputDimensions );
 for(UINT i=0; i<numNodes; i++){
 
 //load the nodeID
 file >> nodeID;
 
 for(UINT j=0; j<numInputDimensions; j++){
 file >> cluster[j];
 }
 
 //Add the cluster to the cluster nodes map
 nodeClusters[ nodeID ] = cluster;
 }
 
 //Recompute the null rejection thresholds
 recomputeNullRejectionThresholds();
 }
 
 //Resize the prediction results to make sure it is setup for realtime prediction
 maxLikelihood = DEFAULT_NULL_LIKELIHOOD_VALUE;
 bestDistance = DEFAULT_NULL_DISTANCE_VALUE;
 classLikelihoods.resize(numClasses,DEFAULT_NULL_LIKELIHOOD_VALUE);
 classDistances.resize(numClasses,DEFAULT_NULL_DISTANCE_VALUE);
 }
 
 return true;
}

GRT_END_NAMESPACE