diff --git a/DESCRIPTION b/DESCRIPTION index 3c6feb9ba..7897b2213 100644 --- a/DESCRIPTION +++ b/DESCRIPTION @@ -1,6 +1,6 @@ Package: parsnip Title: A Common API to Modeling and Analysis Functions -Version: 1.0.3.9001 +Version: 1.0.3.9002 Authors@R: c( person("Max", "Kuhn", , "max@rstudio.com", role = c("aut", "cre")), person("Davis", "Vaughan", , "davis@rstudio.com", role = "aut"), diff --git a/R/multinom_reg.R b/R/multinom_reg.R index 64bb10774..5e23ba544 100644 --- a/R/multinom_reg.R +++ b/R/multinom_reg.R @@ -237,12 +237,13 @@ multi_predict._multnet <- names(pred) <- NULL pred <- map_dfr(pred, function(x) x) pred$penalty <- rep(penalty, each = nrow(new_data)) + pred <- dplyr::relocate(pred, penalty) } else { pred <- tibble( .row = rep(1:nrow(new_data), length(penalty)), - .pred_class = factor(as.vector(pred), levels = object$lvl), - penalty = rep(penalty, each = nrow(new_data)) + penalty = rep(penalty, each = nrow(new_data)), + .pred_class = factor(as.vector(pred), levels = object$lvl) ) }