GitHub user WeichenXu123 opened a pull request:

    [SPARK-14681][ML] Provide label/impurity stats for decision tree 

    ## What changes were proposed in this pull request?
    Provide label/impurity stats for decision tree nodes.
    class TreeClassifierStatInfo
       def getLabelCount(label: Int): Double
    class TreeRegressorStatInfo
       def getCount(): Double
       def getSum(): Double
       def getSquareSum(): Double
    class Node
       +++ def statInfo: TreeStatInfo
    trait TreeStatInfo
       def asTreeClassifierStatInfo: TreeClassifierStatInfo
       def asTreeRegressorStatInfo: TreeRegressorStatInfo
    ## How was this patch tested?
    UT added.

You can merge this pull request into a Git repository by running:

    $ git pull tree_stat_api

Alternatively you can review and apply these changes as the patch at:

To close this pull request, make a commit to your master/trunk branch
with (at least) the following in the commit message:

    This closes #20758
commit e57ffaaad1666577d956c1f8f734f97569b93969
Author: WeichenXu <weichen.xu@...>
Date:   2018-03-07T10:37:22Z

    init pr



To unsubscribe, e-mail:
For additional commands, e-mail:

Reply via email to