added files for regularized random forest

[mothur.git] / regularizedrandomforest.h

diff --git a/regularizedrandomforest.h b/regularizedrandomforest.h
new file mode 100644 (file)
index 0000000..94bd624
--- /dev/null
+++ b/regularizedrandomforest.h
@@ -0,0 +1,30 @@
+//
+//  regularizedrandomforest.h
+//  Mothur
+//
+//  Created by Kathryn Iverson on 11/16/12.
+//  Copyright (c) 2012 Schloss Lab. All rights reserved.
+//
+
+#ifndef __Mothur__regularizedrandomforest__
+#define __Mothur__regularizedrandomforest__
+
+#include "forest.h"
+#include "decisiontree.hpp"
+
+class RegularizedRandomForest: public Forest {
+public:
+    //
+    RegularizedRandomForest(const vector <vector<int> > dataSet,const int numDecisionTrees, const string);
+    
+    int calcForrestErrorRate();
+    int calcForrestVariableImportance(string);
+    int populateDecisionTrees();
+    int updateGlobalOutOfBagEstimates(DecisionTree* decisionTree);
+    
+private:
+    //
+    MothurOut* m;
+};
+
+#endif /* defined(__Mothur__regularizedrandomforest__) */