-
- Downloads
[SPARK-12566][SPARK-14324][ML] GLM model family, link function support in SparkR:::glm
* SparkR glm supports families and link functions which match R's signature for family. * SparkR glm API refactor. The comparative standard of the new API is R glm, so I only expose the arguments that R glm supports: ```formula, family, data, epsilon and maxit```. * This PR is focus on glm() and predict(), summary statistics will be done in a separate PR after this get in. * This PR depends on #12287 which make GLMs support link prediction at Scala side. After that merged, I will add more tests for predict() to this PR. Unit tests. cc mengxr jkbradley hhbyyh Author: Yanbo Liang <ybliang8@gmail.com> Closes #12294 from yanboliang/spark-12566.
Showing
- R/pkg/R/mllib.R 61 additions, 78 deletionsR/pkg/R/mllib.R
- R/pkg/inst/tests/testthat/test_mllib.R 29 additions, 66 deletionsR/pkg/inst/tests/testthat/test_mllib.R
- mllib/src/main/scala/org/apache/spark/ml/r/GeneralizedLinearRegressionWrapper.scala 79 additions, 0 deletions...pache/spark/ml/r/GeneralizedLinearRegressionWrapper.scala
- mllib/src/main/scala/org/apache/spark/ml/r/SparkRWrappers.scala 0 additions, 115 deletions...src/main/scala/org/apache/spark/ml/r/SparkRWrappers.scala
Loading
Please register or sign in to comment