speed up format_classprobs() helper (#934)

simonpcouch · web-flow · commit dfa7d9ac7364 · 2023-03-21T14:33:17.000-04:00
diff --git a/R/predict.R b/R/predict.R
@@ -258,8 +258,14 @@ 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)
+  if (!tibble::is_tibble(x)) {
+    x <- as_tibble(x)
+  }
+
+  for (i in seq_along(x)) {
+    names(x[[i]]) <- NULL
+  }
+
   x
 }
 

Original file line number	Diff line number	Diff line change
`@@ -258,8 +258,14 @@ format_classprobs <- function(x) {`
`258`	`258`	`if (!any(grepl("^\\.pred_", names(x)))) {`
`259`	`259`	`names(x) <- paste0(".pred_", names(x))`
`260`	`260`	`}`
`261`		`- x <- as_tibble(x)`
`262`		`- x <- purrr::map_dfr(x, rlang::set_names, NULL)`
	`261`	`+ if (!tibble::is_tibble(x)) {`
	`262`	`+ x <- as_tibble(x)`
	`263`	`+ }`
	`264`	`+`
	`265`	`+ for (i in seq_along(x)) {`
	`266`	`+ names(x[[i]]) <- NULL`
	`267`	`+ }`
	`268`	`+`
`263`	`269`	`x`
`264`	`270`	`}`
`265`	`271`