Skip to content

Commit e15d7ea

Browse files
committed
Change decode method
1 parent 2928da0 commit e15d7ea

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

R/mlPrediction.R

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -259,7 +259,8 @@ is.jaspMachineLearning <- function(x) {
259259
return(model)
260260
}
261261
.decodeJaspMLobject.glm <- function(model) {
262-
formula <- formula(paste(decodeColNames(as.character(model$terms)[2]), "~", paste0(decodeColNames(strsplit(as.character(model$terms)[3], split = " + ", fixed = TRUE)[[1]]), collapse = " + ")))
262+
vars <- all.vars(stats::terms(model))
263+
formula <- formula(paste(decodeColNames(vars[1]), "~", paste0(decodeColNames(vars[-1]), collapse = " + ")))
263264
model$terms <- stats::terms(formula)
264265
return(model)
265266
}

tests/testthat/test-mlclassificationlogisticmultinomial.R

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ options$coefTable <- TRUE
88
options$coefTableConfInt <- TRUE
99
options$classProportionsTable <- TRUE
1010
options$holdoutData <- "holdoutManual"
11+
options$link <- "logit"
1112
options$modelOptimization <- "manual"
1213
options$modelValid <- "validationManual"
1314
options$predictionsColumn <- ""
@@ -107,6 +108,7 @@ options$coefTable <- TRUE
107108
options$coefTableConfInt <- TRUE
108109
options$classProportionsTable <- TRUE
109110
options$holdoutData <- "holdoutManual"
111+
options$link <- "logit"
110112
options$modelOptimization <- "manual"
111113
options$modelValid <- "validationManual"
112114
options$predictionsColumn <- ""

0 commit comments

Comments
 (0)