[
https://issues.apache.org/jira/browse/SPARK-6885?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14642568#comment-14642568
]
Yanbo Liang commented on SPARK-6885:
------------------------------------
[~josephkb]
I referred the old DecisionTree API and the sklean API
(https://github.com/scikit-learn/scikit-learn/blob/master/sklearn/tree/tree.py#L593)
to implement the prediction probabilities function.
I make the DecisionTreeClassificationModel inherit from
ProbabilisticClassificationModel, make the predictRaw to return the
probabilities and make raw2probabilityInPlace just return the rawPrediction.
Any comments?
Another issue is the "numClasses" variables which is not tackled appropriately
at present, I think the numClasses should become one of the ClassifierParams
and can be set by Classifier. I will optimize this issue after collecting
comments.
> Decision trees: predict class probabilities
> -------------------------------------------
>
> Key: SPARK-6885
> URL: https://issues.apache.org/jira/browse/SPARK-6885
> Project: Spark
> Issue Type: Sub-task
> Components: ML
> Affects Versions: 1.3.0
> Reporter: Joseph K. Bradley
>
> Under spark.ml, have DecisionTreeClassifier (currently being added) extend
> ProbabilisticClassifier.
--
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
