Skip to content

Commit 0c8893d

Browse files
authored
Merge pull request #864 from tidymodels/glmnet-multi_predict-column-order
Switch column order of `multi_predict()` result for `multinom_reg(engine = "glmnet")`
2 parents a517c87 + 2c704e9 commit 0c8893d

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

DESCRIPTION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
Package: parsnip
22
Title: A Common API to Modeling and Analysis Functions
3-
Version: 1.0.3.9001
3+
Version: 1.0.3.9002
44
Authors@R: c(
55
person("Max", "Kuhn", , "max@rstudio.com", role = c("aut", "cre")),
66
person("Davis", "Vaughan", , "davis@rstudio.com", role = "aut"),

R/multinom_reg.R

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -237,12 +237,13 @@ multi_predict._multnet <-
237237
names(pred) <- NULL
238238
pred <- map_dfr(pred, function(x) x)
239239
pred$penalty <- rep(penalty, each = nrow(new_data))
240+
pred <- dplyr::relocate(pred, penalty)
240241
} else {
241242
pred <-
242243
tibble(
243244
.row = rep(1:nrow(new_data), length(penalty)),
244-
.pred_class = factor(as.vector(pred), levels = object$lvl),
245-
penalty = rep(penalty, each = nrow(new_data))
245+
penalty = rep(penalty, each = nrow(new_data)),
246+
.pred_class = factor(as.vector(pred), levels = object$lvl)
246247
)
247248
}
248249

0 commit comments

Comments
 (0)