// Copyright (c) 2012 Schloss Lab. All rights reserved.
//
-#ifndef rrf_fs_prototype_treenode_hpp
-#define rrf_fs_prototype_treenode_hpp
+#ifndef RF_RFTREENODE_HPP
+#define RF_RFTREENODE_HPP
#include "mothurout.h"
#include "macros.h"
public:
- RFTreeNode(vector< vector<int> > bootstrappedTrainingSamples, vector<int> globalDiscardedFeatureIndices, int numFeatures, int numSamples, int numOutputClasses, int generation);
+ RFTreeNode(vector< vector<int> > bootstrappedTrainingSamples,
+ vector<int> globalDiscardedFeatureIndices,
+ int numFeatures,
+ int numSamples,
+ int numOutputClasses,
+ int generation,
+ int nodeId,
+ float featureStandardDeviationThreshold = 0.0);
virtual ~RFTreeNode(){}
const vector<int>& getBootstrappedOutputVector() { return bootstrappedOutputVector; }
const vector<int>& getFeatureSubsetIndices() { return featureSubsetIndices; }
const double getOwnEntropy() { return ownEntropy; }
+ const int getTestSampleMisclassificationCount() { return testSampleMisclassificationCount; }
// setters
void setIsLeaf(bool isLeaf) { this->isLeaf = isLeaf; }
double splitFeatureEntropy;
double ownEntropy;
+ int nodeId;
+ float featureStandardDeviationThreshold;
+ int testSampleMisclassificationCount;
+
RFTreeNode* leftChildNode;
RFTreeNode* rightChildNode;
RFTreeNode* parentNode;