[ https://issues.apache.org/jira/browse/SPARK-8335?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14589492#comment-14589492 ]
Apache Spark commented on SPARK-8335: ------------------------------------- User 'srowen' has created a pull request for this issue: https://github.com/apache/spark/pull/6854 > 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