X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=randomforest.hpp;h=d0ac1ec063d46d4ff1367e87cf322870f7e49a9c;hb=541bab1dac00688b4c3a8c4a95ab464412663c50;hp=716d1a1ede60c667840172db275814c73867041a;hpb=035f86272c776e1cccaa47021e26782e49cd41e7;p=mothur.git diff --git a/randomforest.hpp b/randomforest.hpp old mode 100755 new mode 100644 index 716d1a1..d0ac1ec --- a/randomforest.hpp +++ b/randomforest.hpp @@ -6,19 +6,26 @@ // Copyright (c) 2012 Schloss Lab. All rights reserved. // -#ifndef rrf_fs_prototype_randomforest_hpp -#define rrf_fs_prototype_randomforest_hpp +#ifndef RF_RANDOMFOREST_HPP +#define RF_RANDOMFOREST_HPP #include "macros.h" -#include "abstractrandomforest.hpp" +#include "forest.h" #include "decisiontree.hpp" -class RandomForest: public AbstractRandomForest { +class RandomForest: public Forest { public: - // DONE - RandomForest(const vector > dataSet,const int numDecisionTrees, const string); + RandomForest(const vector > dataSet, + const int numDecisionTrees, + const string treeSplitCriterion, + const bool doPruning, + const float pruneAggressiveness, + const bool discardHighErrorTrees, + const float highErrorTreeDiscardThreshold, + const string optimumFeatureSubsetSelectionCriteria, + const float featureStandardDeviationThreshold); //NOTE:: if you are going to dynamically cast, aren't you undoing the advantage of abstraction. Why abstract at all?