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

Examples of org.apache.spark.mllib.tree.configuration.Strategy

68
69
70
71
72
73
74
75
76
77
78
    categoricalFeaturesInfo.put(1, 2); // feature 1 has 2 categories

    int maxDepth = 4;
    int numClasses = 2;
    int maxBins = 100;
    Strategy strategy = new Strategy(Algo.Classification(), Gini.instance(), maxDepth, numClasses,
        maxBins, categoricalFeaturesInfo);

    DecisionTree learner = new DecisionTree(strategy);
    DecisionTreeModel model = learner.train(rdd.rdd());
View Full Code Here
88
89
90
91
92
93
94
95
96
97
98
    categoricalFeaturesInfo.put(1, 2); // feature 1 has 2 categories

    int maxDepth = 4;
    int numClasses = 2;
    int maxBins = 100;
    Strategy strategy = new Strategy(Algo.Classification(), Gini.instance(), maxDepth, numClasses,
        maxBins, categoricalFeaturesInfo);

    DecisionTreeModel model = DecisionTree$.MODULE$.train(rdd.rdd(), strategy);

    int numCorrect = validatePrediction(arr, model);
View Full Code Here
TOP

Related Classes of org.apache.spark.mllib.tree.configuration.Strategy

  • org.apache.spark.mllib.tree.JavaDecisionTreeSuite
    • 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.