-
- Downloads
[SPARK-10524][ML] Use the soft prediction to order categories' bins
JIRA: https://issues.apache.org/jira/browse/SPARK-10524 Currently we use the hard prediction (`ImpurityCalculator.predict`) to order categories' bins. But we should use the soft prediction. Author: Liang-Chi Hsieh <viirya@gmail.com> Author: Liang-Chi Hsieh <viirya@appier.com> Author: Joseph K. Bradley <joseph@databricks.com> Closes #8734 from viirya/dt-soft-centroids.
Showing
- mllib/src/main/scala/org/apache/spark/ml/tree/impl/RandomForest.scala 20 additions, 22 deletions...in/scala/org/apache/spark/ml/tree/impl/RandomForest.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala 109 additions, 110 deletions...main/scala/org/apache/spark/mllib/tree/DecisionTree.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/DecisionTreeClassifierSuite.scala 35 additions, 1 deletion...spark/ml/classification/DecisionTreeClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala 30 additions, 0 deletions...scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala
Loading
Please register or sign in to comment