protected LearningModule getLearningModule(String name) throws Exception {
int maxNumTrainInstances = trainDataset.numInstances;
int maxNumValidInstances = (validDataset != null ? validDataset.numInstances : trainDataset.numInstances);
if (name.equals("GradientBoostingBinaryClassifier")) {
GradientBoostingBinaryClassifier learner = new GradientBoostingBinaryClassifier();
learner.init(configHolder, maxNumTrainInstances, maxNumValidInstances, evaluationMetric);
return learner;
} else if (name.equals("GradientBoosting")) {
GradientBoosting learner = new GradientBoosting();
learner.init(configHolder, maxNumTrainInstances, maxNumValidInstances, evaluationMetric);
return learner;
} else if (name.equals("RegressionTree")) {
RegressionTreeLearner learner = new RegressionTreeLearner();
learner.init(trainDataset, configHolder, maxNumTrainInstances);
return learner;
} else if (name.equals("RandomForest")) {
RandomForest learner = new RandomForest();
learner.init(trainDataset, configHolder, maxNumTrainInstances, maxNumValidInstances, evaluationMetric);
return learner;
} else {
throw new Exception("Unknown algorithm: " + name);
}
}