diff options
author | Yanbo Liang <ybliang8@gmail.com> | 2015-10-28 08:50:21 -0700 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-10-28 08:50:21 -0700 |
commit | fba9e95452ca0a9b589bc14b27c750c69f482b8d (patch) | |
tree | 944f386177e78ac7d567016d9feb3e3e33f78e01 /R | |
parent | fd9e345ceeff385ba614a16d478097650caa98d0 (diff) | |
download | spark-fba9e95452ca0a9b589bc14b27c750c69f482b8d.tar.gz spark-fba9e95452ca0a9b589bc14b27c750c69f482b8d.tar.bz2 spark-fba9e95452ca0a9b589bc14b27c750c69f482b8d.zip |
[SPARK-11369][ML][R] SparkR glm should support setting standardize
SparkR glm currently support :
```formula, family = c(“gaussian”, “binomial”), data, lambda = 0, alpha = 0```
We should also support setting standardize which has been defined at [design documentation](https://docs.google.com/document/d/10NZNSEurN2EdWM31uFYsgayIPfCFHiuIu3pCWrUmP_c/edit)
Author: Yanbo Liang <ybliang8@gmail.com>
Closes #9331 from yanboliang/spark-11369.
Diffstat (limited to 'R')
-rw-r--r-- | R/pkg/R/mllib.R | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/R/pkg/R/mllib.R b/R/pkg/R/mllib.R index 25615e805e..aadd5b8da5 100644 --- a/R/pkg/R/mllib.R +++ b/R/pkg/R/mllib.R @@ -46,11 +46,11 @@ setClass("PipelineModel", representation(model = "jobj")) #'} setMethod("glm", signature(formula = "formula", family = "ANY", data = "DataFrame"), function(formula, family = c("gaussian", "binomial"), data, lambda = 0, alpha = 0, - solver = "auto") { + standardize = TRUE, solver = "auto") { family <- match.arg(family) model <- callJStatic("org.apache.spark.ml.api.r.SparkRWrappers", "fitRModelFormula", deparse(formula), data@sdf, family, lambda, - alpha, solver) + alpha, standardize, solver) return(new("PipelineModel", model = model)) }) |