factor_cols fix

bips-hb · Aug 30, 2023 · 8077c7e · 8077c7e
1 parent 2710517
commit 8077c7e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/R/forde.R b/R/forde.R
@@ -114,10 +114,11 @@ forde <- function(
   x <- as.data.frame(x)
   n <- nrow(x)
   d <- ncol(x)
-  factor_cols <- sapply(x, is.factor)
   colnames_x <- colnames(x)
   classes <- sapply(x, class)
   x <- suppressWarnings(prep_x(x))
+  factor_cols <- sapply(x, is.factor)
+  names(factor_cols) <- colnames_x
 
   # Compute leaf bounds and coverage
   num_trees <- arf$num.trees