Sebastian Walz created SPARK-8335:
-------------------------------------
Summary: 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
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 from the return type JAVARDD[Double]
because its a scala Double and and java.lang.Double like expected.
I wanted to extend the DecisionTreeModel and use it only for Binary
Classification and wanted to extend with 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
