-
- Downloads
[SPARK-18326][SPARKR][ML] Review SparkR ML wrappers API for 2.1
## What changes were proposed in this pull request? Reviewing SparkR ML wrappers API for 2.1 release, mainly two issues: * Remove ```probabilityCol``` from the argument list of ```spark.logit``` and ```spark.randomForest```. Since it was used when making prediction and should be an argument of ```predict```, and we will work on this at [SPARK-18618](https://issues.apache.org/jira/browse/SPARK-18618 ) in the next release cycle. * Fix ```spark.als``` params to make it consistent with MLlib. ## How was this patch tested? Existing tests. Author: Yanbo Liang <ybliang8@gmail.com> Closes #16169 from yanboliang/spark-18326. (cherry picked from commit 97255497) Signed-off-by:Yanbo Liang <ybliang8@gmail.com>
Showing
- R/pkg/R/mllib.R 10 additions, 13 deletionsR/pkg/R/mllib.R
- R/pkg/inst/tests/testthat/test_mllib.R 2 additions, 2 deletionsR/pkg/inst/tests/testthat/test_mllib.R
- mllib/src/main/scala/org/apache/spark/ml/r/LogisticRegressionWrapper.scala 1 addition, 3 deletions...ala/org/apache/spark/ml/r/LogisticRegressionWrapper.scala
- mllib/src/main/scala/org/apache/spark/ml/r/RandomForestClassificationWrapper.scala 0 additions, 2 deletions...apache/spark/ml/r/RandomForestClassificationWrapper.scala
Loading
Please register or sign in to comment