From 2c8f925a31fb9bb7a178e71d572397da1ba03923 Mon Sep 17 00:00:00 2001 From: Sean Owen Date: Wed, 17 Jun 2015 09:36:19 +0100 Subject: [PATCH] Make Java-friendly predict return a JavaRDD of java.lang.Double instead of Scala Double, in the way ClassificationModel et al do --- .../org/apache/spark/mllib/tree/model/DecisionTreeModel.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/model/DecisionTreeModel.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/model/DecisionTreeModel.scala index 25bb1453db404..2ee6d45c75a6d 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/tree/model/DecisionTreeModel.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/tree/model/DecisionTreeModel.scala @@ -70,8 +70,8 @@ class DecisionTreeModel(val topNode: Node, val algo: Algo) extends Serializable * @param features JavaRDD representing data points to be predicted * @return JavaRDD of predictions for each of the given data points */ - def predict(features: JavaRDD[Vector]): JavaRDD[Double] = { - predict(features.rdd) + def predict(features: JavaRDD[Vector]): JavaRDD[java.lang.Double] = { + predict(features.rdd).toJavaRDD().asInstanceOf[JavaRDD[java.lang.Double]] } /**