-
- Downloads
[SPARK-18239][SPARKR] Gradient Boosted Tree for R
## What changes were proposed in this pull request? Gradient Boosted Tree in R. With a few minor improvements to RandomForest in R. Since this is relatively isolated I'd like to target this for branch-2.1 ## How was this patch tested? manual tests, unit tests Author: Felix Cheung <felixcheung_m@hotmail.com> Closes #15746 from felixcheung/rgbt. (cherry picked from commit 55964c15) Signed-off-by:Felix Cheung <felixcheung@apache.org>
Showing
- R/pkg/NAMESPACE 7 additions, 2 deletionsR/pkg/NAMESPACE
- R/pkg/R/generics.R 4 additions, 0 deletionsR/pkg/R/generics.R
- R/pkg/R/mllib.R 286 additions, 45 deletionsR/pkg/R/mllib.R
- R/pkg/inst/tests/testthat/test_mllib.R 68 additions, 0 deletionsR/pkg/inst/tests/testthat/test_mllib.R
- mllib/src/main/scala/org/apache/spark/ml/r/GBTClassificationWrapper.scala 164 additions, 0 deletions...cala/org/apache/spark/ml/r/GBTClassificationWrapper.scala
- mllib/src/main/scala/org/apache/spark/ml/r/GBTRegressionWrapper.scala 144 additions, 0 deletions...in/scala/org/apache/spark/ml/r/GBTRegressionWrapper.scala
- mllib/src/main/scala/org/apache/spark/ml/r/RWrappers.scala 4 additions, 0 deletionsmllib/src/main/scala/org/apache/spark/ml/r/RWrappers.scala
- mllib/src/main/scala/org/apache/spark/ml/r/RandomForestClassificationWrapper.scala 7 additions, 7 deletions...apache/spark/ml/r/RandomForestClassificationWrapper.scala
- mllib/src/main/scala/org/apache/spark/ml/r/RandomForestRegressionWrapper.scala 7 additions, 7 deletions...org/apache/spark/ml/r/RandomForestRegressionWrapper.scala
- python/pyspark/ml/regression.py 5 additions, 5 deletionspython/pyspark/ml/regression.py
Loading
Please register or sign in to comment