X-Git-Url: https://git.donarmstrong.com/?p=mothur.git;a=blobdiff_plain;f=regularizedrandomforest.cpp;h=0a33f4e3ba6ed2a79a301c2db5c8254ae153784c;hp=55a8dc4a4284d07aef4211504e648be50a6cbc78;hb=050a3ff02473a3d4c0980964e1a9ebe52e55d6b8;hpb=deba0af0ccdcb6005ed5b2b82649b137c63fbdf7 diff --git a/regularizedrandomforest.cpp b/regularizedrandomforest.cpp index 55a8dc4..0a33f4e 100644 --- a/regularizedrandomforest.cpp +++ b/regularizedrandomforest.cpp @@ -8,8 +8,14 @@ #include "regularizedrandomforest.h" -RegularizedRandomForest::RegularizedRandomForest(const vector > dataSet,const int numDecisionTrees, - const string treeSplitCriterion = "informationGain") : Forest(dataSet, numDecisionTrees, treeSplitCriterion) { +RegularizedRandomForest::RegularizedRandomForest(const vector > dataSet, + const int numDecisionTrees, + const string treeSplitCriterion = "gainratio") + // TODO: update ctor according to basic RandomForest Class + : Forest(dataSet, + numDecisionTrees, + treeSplitCriterion, + false, 0.9, true, 0.4, "log2", 0.0) { m = MothurOut::getInstance(); } @@ -54,4 +60,4 @@ int RegularizedRandomForest::updateGlobalOutOfBagEstimates(DecisionTree *decisio m->errorOut(e, "RegularizedRandomForest", "updateGlobalOutOfBagEstimates"); exit(1); } -} \ No newline at end of file +}