Skip to content

Commit c8bdd3b

Browse files
committed
[SPARK-17241] change param order
1 parent 6417049 commit c8bdd3b

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

R/pkg/R/mllib.R

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -172,8 +172,8 @@ predict_internal <- function(object, newData) {
172172
#' @note spark.glm since 2.0.0
173173
#' @seealso \link{glm}, \link{read.ml}
174174
setMethod("spark.glm", signature(data = "SparkDataFrame", formula = "formula"),
175-
function(data, formula, family = gaussian, tol = 1e-6, regParam = 0.0, maxIter = 25,
176-
weightCol = NULL) {
175+
function(data, formula, family = gaussian, tol = 1e-6, maxIter = 25, weightCol = NULL,
176+
regParam = 0.0) {
177177
if (is.character(family)) {
178178
family <- get(family, mode = "function", envir = parent.frame())
179179
}
@@ -192,7 +192,7 @@ setMethod("spark.glm", signature(data = "SparkDataFrame", formula = "formula"),
192192

193193
jobj <- callJStatic("org.apache.spark.ml.r.GeneralizedLinearRegressionWrapper",
194194
"fit", formula, data@sdf, family$family, family$link,
195-
tol, regParam, as.integer(maxIter), as.character(weightCol))
195+
tol, as.integer(maxIter), as.character(weightCol), regParam)
196196
new("GeneralizedLinearRegressionModel", jobj = jobj)
197197
})
198198

mllib/src/main/scala/org/apache/spark/ml/r/GeneralizedLinearRegressionWrapper.scala

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,9 +68,9 @@ private[r] object GeneralizedLinearRegressionWrapper
6868
family: String,
6969
link: String,
7070
tol: Double,
71-
regParam: Double,
7271
maxIter: Int,
73-
weightCol: String): GeneralizedLinearRegressionWrapper = {
72+
weightCol: String,
73+
regParam: Double): GeneralizedLinearRegressionWrapper = {
7474
val rFormula = new RFormula()
7575
.setFormula(formula)
7676
val rFormulaModel = rFormula.fit(data)
@@ -85,9 +85,9 @@ private[r] object GeneralizedLinearRegressionWrapper
8585
.setLink(link)
8686
.setFitIntercept(rFormula.hasIntercept)
8787
.setTol(tol)
88-
.setRegParam(regParam)
8988
.setMaxIter(maxIter)
9089
.setWeightCol(weightCol)
90+
.setRegParam(regParam)
9191
val pipeline = new Pipeline()
9292
.setStages(Array(rFormulaModel, glr))
9393
.fit(data)

0 commit comments

Comments
 (0)