Package edu.uci.jforestsx.learning.classification

Examples of edu.uci.jforestsx.learning.classification.GradientBoostingBinaryClassifier


  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);
    }
  }
View Full Code Here

TOP

Related Classes of edu.uci.jforestsx.learning.classification.GradientBoostingBinaryClassifier

Copyright © 2018 www.massapicom. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.