[
https://issues.apache.org/jira/browse/SPARK-6885?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14647741#comment-14647741
]
Yanbo Liang commented on SPARK-6885:
------------------------------------
[~josephkb]
I create a new version of InformationGainStats called ImpurityStats. It stores
information gain, impurity, prediction related data all in one data structure
which make LearningNode simplicity. Meanwhile it simplifies and optimizes
binsToBestSplit function.
I will fix some trivial issues after your reviews. It looks like code refactor
in a way.
> 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
> Assignee: Yanbo Liang
>
> 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
