Package org.apache.spark.mllib.tree.configuration

Examples of org.apache.spark.mllib.tree.configuration.BoostingStrategy.weakLearnerParams()

62
63
64
65
66
67
68
69
70
71
72
    // Set parameters.
    //  Note: All features are treated as continuous.
    BoostingStrategy boostingStrategy = BoostingStrategy.defaultParams(algo);
    boostingStrategy.setNumIterations(10);
    boostingStrategy.weakLearnerParams().setMaxDepth(5);

    if (algo.equals("Classification")) {
      // Compute the number of classes from the data.
      Integer numClasses = data.map(new Function<LabeledPoint, Double>() {
        @Override public Double call(LabeledPoint p) {
View Full Code Here
TOP
Copyright © 2018 www.massapi.com. 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.