[ https://issues.apache.org/jira/browse/SPARK-8335?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14591393#comment-14591393 ]
Joseph K. Bradley commented on SPARK-8335: ------------------------------------------ My initial thought was to fix it to make it consistent, but I was discouraged from making a breaking change for a nicety like this. I think Matei & some of the original people in particular feel strongly about maintaining stable APIs as much as possible, even if items are marked as experimental/developer api. > 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