-
- Downloads
[SPARK-19282][ML][SPARKR] RandomForest Wrapper and GBT Wrapper return param "maxDepth" to R models
## What changes were proposed in this pull request? RandomForest R Wrapper and GBT R Wrapper return param `maxDepth` to R models. Below 4 R wrappers are changed: * `RandomForestClassificationWrapper` * `RandomForestRegressionWrapper` * `GBTClassificationWrapper` * `GBTRegressionWrapper` ## How was this patch tested? Test manually on my local machine. Author: Xin Ren <iamshrek@126.com> Closes #17207 from keypointt/SPARK-19282.
Showing
- R/pkg/R/mllib_tree.R 7 additions, 4 deletionsR/pkg/R/mllib_tree.R
- R/pkg/inst/tests/testthat/test_mllib_tree.R 10 additions, 0 deletionsR/pkg/inst/tests/testthat/test_mllib_tree.R
- mllib/src/main/scala/org/apache/spark/ml/r/GBTClassificationWrapper.scala 1 addition, 0 deletions...cala/org/apache/spark/ml/r/GBTClassificationWrapper.scala
- mllib/src/main/scala/org/apache/spark/ml/r/GBTRegressionWrapper.scala 1 addition, 0 deletions...in/scala/org/apache/spark/ml/r/GBTRegressionWrapper.scala
- mllib/src/main/scala/org/apache/spark/ml/r/RandomForestClassificationWrapper.scala 1 addition, 0 deletions...apache/spark/ml/r/RandomForestClassificationWrapper.scala
- mllib/src/main/scala/org/apache/spark/ml/r/RandomForestRegressionWrapper.scala 1 addition, 0 deletions...org/apache/spark/ml/r/RandomForestRegressionWrapper.scala
Loading
Please register or sign in to comment