ensemble model fix

microsoft · Oct 14, 2024 · d3dbf41 · d3dbf41
1 parent 5c79652
commit d3dbf41
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/R/ensemble_models.R b/R/ensemble_models.R
@@ -236,7 +236,8 @@ ensemble_models <- function(run_info,
         avail_arg_list <- list(
           "train_data" = prep_ensemble_tbl %>% dplyr::select(-Train_Test_ID),
           "model_type" = "ensemble",
-          "pca" = FALSE
+          "pca" = FALSE, 
+          "multistep" = FALSE
         )
 
         # get specific model spec