-
- Downloads
[SPARK-8736] [ML] GBTRegressor should not threshold prediction
Changed GBTRegressor so it does NOT threshold the prediction. Added test which fails with bug but works after fix. CC: feynmanliang mengxr Author: Joseph K. Bradley <joseph@databricks.com> Closes #7134 from jkbradley/gbrt-fix and squashes the following commits: 613b90e [Joseph K. Bradley] Changed GBTRegressor so it does NOT threshold the prediction
Showing
- mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala 1 addition, 2 deletions...n/scala/org/apache/spark/ml/regression/GBTRegressor.scala
- mllib/src/test/scala/org/apache/spark/ml/regression/GBTRegressorSuite.scala 22 additions, 1 deletion...la/org/apache/spark/ml/regression/GBTRegressorSuite.scala
Please register or sign in to comment