Update 2-modeling.R

This commit is contained in:
Kyle Belanger 2023-06-05 17:13:19 -04:00
parent a5fc4dd7d6
commit 3251c95310

View file

@ -59,7 +59,7 @@ rf_recipe <- r$recipe(ft4_dia ~ . , data = ds_train) %>%
r$step_impute_bag(r$all_predictors())
rf_tuning_model <- p$rand_forest(trees = tune(), mtry = tune(), min_n = tune()) %>%
p$set_engine("ranger") %>% p$set_mode("classification")
p$set_engine("ranger", importance = "impurity") %>% p$set_mode("classification")
rf_workflow <- wf$workflow() %>%