public class SparkDecisionTreeModel extends Object implements org.apache.spark.mllib.classification.ClassificationModel
Constructor and Description |
---|
SparkDecisionTreeModel(org.apache.spark.mllib.tree.model.DecisionTreeModel model)
Instantiates a new spark decision tree model.
|
Modifier and Type | Method and Description |
---|---|
org.apache.spark.api.java.JavaRDD<Double> |
predict(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.linalg.Vector> testData) |
org.apache.spark.rdd.RDD<Object> |
predict(org.apache.spark.rdd.RDD<org.apache.spark.mllib.linalg.Vector> testData) |
double |
predict(org.apache.spark.mllib.linalg.Vector testData) |
public SparkDecisionTreeModel(org.apache.spark.mllib.tree.model.DecisionTreeModel model)
model
- the modelpublic org.apache.spark.rdd.RDD<Object> predict(org.apache.spark.rdd.RDD<org.apache.spark.mllib.linalg.Vector> testData)
predict
in interface org.apache.spark.mllib.classification.ClassificationModel
public double predict(org.apache.spark.mllib.linalg.Vector testData)
predict
in interface org.apache.spark.mllib.classification.ClassificationModel
public org.apache.spark.api.java.JavaRDD<Double> predict(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.linalg.Vector> testData)
predict
in interface org.apache.spark.mllib.classification.ClassificationModel
Copyright © 2014–2018 Apache Software Foundation. All rights reserved.