[
https://issues.apache.org/jira/browse/SPARK-8335?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14584481#comment-14584481
]
Sean Owen commented on SPARK-8335:
----------------------------------
Yeah, the signature is actually a JavaRDD of scala.Double given how it's
declared. Just changing the type to java.lang.Double probably invokes the right
implicits to convert it, as happens elsewhere. You've verified it really is a
scala.Double?
> DecisionTreeModel.predict() return type not convenient!
> -------------------------------------------------------
>
> Key: SPARK-8335
> URL: https://issues.apache.org/jira/browse/SPARK-8335
> Project: Spark
> Issue Type: Bug
> Components: MLlib
> Affects Versions: 1.3.1
> Reporter: Sebastian Walz
> Priority: Minor
> Labels: easyfix, machine_learning
> Original Estimate: 10m
> Remaining Estimate: 10m
>
> org.apache.spark.mllib.tree.model.DecisionTreeModel has a predict method:
> def predict(features: JavaRDD[Vector]): JavaRDD[Double]
> The problem here is the generic type of the return type JAVARDD[Double]
> because its a scala Double and I would expect a java.lang.Double. (to be
> convenient e.g. with
> org.apache.spark.mllib.classification.ClassificationModel)
> I wanted to extend the DecisionTreeModel and use it only for Binary
> Classification and wanted to implement the trait
> org.apache.spark.mllib.classification.ClassificationModel . But its not
> possible because the ClassificationModel already defines the predict method
> but with an return type JAVARDD[java.lang.Double].
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
