Reorder new_data columns for glmnet

juliasilge · juliasilge · commit c98a44122cdc · 2020-10-20T17:22:50.000-06:00
diff --git a/R/linear_reg_data.R b/R/linear_reg_data.R
@@ -168,7 +168,7 @@ set_pred(
     args =
       list(
         object = expr(object$fit),
-        newx = expr(as.matrix(new_data)),
+        newx = quote(as.matrix(new_data[, rownames(object$fit$beta)])),
         type = "response",
         s = expr(object$spec$args$penalty)
       )
diff --git a/R/logistic_reg_data.R b/R/logistic_reg_data.R
@@ -187,7 +187,7 @@ set_pred(
     args =
       list(
         object = quote(object$fit),
-        newx = quote(as.matrix(new_data)),
+        newx = quote(as.matrix(new_data[, rownames(object$fit$beta)])),
         type = "response",
         s = quote(object$spec$args$penalty)
       )

Original file line number	Diff line number	Diff line change
`@@ -168,7 +168,7 @@ set_pred(`
`168`	`168`	`args =`
`169`	`169`	`list(`
`170`	`170`	`object = expr(object$fit),`
`171`		`- newx = expr(as.matrix(new_data)),`
	`171`	`+ newx = quote(as.matrix(new_data[, rownames(object$fit$beta)])),`
`172`	`172`	`type = "response",`
`173`	`173`	`s = expr(object$spec$args$penalty)`
`174`	`174`	`)`
Original file line number	Diff line number	Diff line change
`@@ -187,7 +187,7 @@ set_pred(`
`187`	`187`	`args =`
`188`	`188`	`list(`
`189`	`189`	`object = quote(object$fit),`
`190`		`- newx = quote(as.matrix(new_data)),`
	`190`	`+ newx = quote(as.matrix(new_data[, rownames(object$fit$beta)])),`
`191`	`191`	`type = "response",`
`192`	`192`	`s = quote(object$spec$args$penalty)`
`193`	`193`	`)`