Example 26 with RegTreeConfig
use of edu.neu.ccs.pyramid.regression.regression_tree.RegTreeConfig in project pyramid by cheng-li.
the class StumpSelector method score.
private static double score(DataSet dataSet, double[] labels) {
RegTreeConfig regTreeConfig = new RegTreeConfig().setMaxNumLeaves(2);
RegressionTree tree = RegTreeTrainer.fit(regTreeConfig, dataSet, labels);
return tree.getRoot().getReduction();
}
Also used :
RegTreeConfig(edu.neu.ccs.pyramid.regression.regression_tree.RegTreeConfig)
RegressionTree(edu.neu.ccs.pyramid.regression.regression_tree.RegressionTree)
Aggregations
RegTreeConfig (edu.neu.ccs.pyramid.regression.regression_tree.RegTreeConfig)26
RegTreeFactory (edu.neu.ccs.pyramid.regression.regression_tree.RegTreeFactory)23
LKBOutputCalculator (edu.neu.ccs.pyramid.classification.lkboost.LKBOutputCalculator)10
LKBoost (edu.neu.ccs.pyramid.classification.lkboost.LKBoost)10
LKBoostOptimizer (edu.neu.ccs.pyramid.classification.lkboost.LKBoostOptimizer)10
RegDataSet (edu.neu.ccs.pyramid.dataset.RegDataSet)9
File (java.io.File)7
LSBoost (edu.neu.ccs.pyramid.regression.least_squares_boost.LSBoost)6
LSBoostOptimizer (edu.neu.ccs.pyramid.regression.least_squares_boost.LSBoostOptimizer)6
DataSetType (edu.neu.ccs.pyramid.dataset.DataSetType)4
RegressionTree (edu.neu.ccs.pyramid.regression.regression_tree.RegressionTree)4
EarlyStopper (edu.neu.ccs.pyramid.optimization.EarlyStopper)3
IOException (java.io.IOException)3
StopWatch (org.apache.commons.lang3.time.StopWatch)3
ClfDataSet (edu.neu.ccs.pyramid.dataset.ClfDataSet)2
ArrayList (java.util.ArrayList)2
List (java.util.List)2
ObjectMapper (com.fasterxml.jackson.databind.ObjectMapper)1
PriorProbClassifier (edu.neu.ccs.pyramid.classification.PriorProbClassifier)1
TRECFormat (edu.neu.ccs.pyramid.dataset.TRECFormat)1
