Merge pull request #313 from tidymodels/inner-names-predict

Remove inner names from prediction output

Author: topepo

R/predict.R

@@ -191,7 +191,7 @@ format_num <- function(x) {
       names(x) <- paste0(".pred_", names(x))
     }
   } else {
-    x <- tibble(.pred = x)
+    x <- tibble(.pred = unname(x))
   }
 
   x
@@ -201,14 +201,15 @@ format_class <- function(x) {
   if (inherits(x, "tbl_spark"))
     return(x)
 
-  tibble(.pred_class = x)
+  tibble(.pred_class = unname(x))
 }
 
 format_classprobs <- function(x) {
   if (!any(grepl("^\\.pred_", names(x)))) {
     names(x) <- paste0(".pred_", names(x))
   }
   x <- as_tibble(x)
+  x <- purrr::map_dfr(x, rlang::set_names, NULL)
   x
 }
 

tests/testthat/test_logistic_reg.R

@@ -300,7 +300,9 @@ test_that('glm probabilities', {
     control = ctrl
   )
 
-  xy_pred <- predict(classes_xy$fit, newdata = lending_club[1:7, num_pred], type = "response")
+  xy_pred <- unname(predict(classes_xy$fit,
+                            newdata = lending_club[1:7, num_pred],
+                            type = "response"))
   xy_pred <- tibble(.pred_bad = 1 - xy_pred, .pred_good = xy_pred)
   expect_equal(xy_pred, predict(classes_xy, lending_club[1:7, num_pred], type = "prob"))
 

tests/testthat/test_logistic_reg_glmnet.R

@@ -251,9 +251,9 @@ test_that('glmnet probabilities, one lambda', {
   form_mat <- form_mat[1:7, -1]
 
   form_pred <-
-    predict(res_form$fit,
+    unname(predict(res_form$fit,
             newx = form_mat,
-            s = 0.1, type = "response")[, 1]
+            s = 0.1, type = "response")[, 1])
   form_pred <- tibble(.pred_bad = 1 - form_pred, .pred_good = form_pred)
 
   expect_equal(
@@ -358,7 +358,8 @@ test_that('glmnet probabilities, no lambda', {
 
   expect_equal(
     mult_pred,
-    multi_predict(xy_fit, lending_club[1:7, num_pred], type = "prob") %>% unnest()
+    multi_predict(xy_fit, lending_club[1:7, num_pred], type = "prob") %>%
+      unnest(cols = c(.pred))
   )
 
   res_form <- fit(