[
https://issues.apache.org/jira/browse/SPARK-8335?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14590922#comment-14590922
]
Joseph K. Bradley commented on SPARK-8335:
------------------------------------------
I've discussed this with [~mengxr] and we decided to leave it alone. I agree
it's annoying, but we figured that people will use the Pipelines API in the
future (where this is not an issue) and not breaking people's code would be
best. Does that sound tolerable?
> 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
