From 3251c9531089a282dced7065030e59e499d5836f Mon Sep 17 00:00:00 2001
From: Kyle Belanger <kyleb44@hotmail.com>
Date: Mon, 5 Jun 2023 17:13:19 -0400
Subject: [PATCH] Update 2-modeling.R

---
 ML/2-modeling.R | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ML/2-modeling.R b/ML/2-modeling.R
index aa05e52..b9735d8 100644
--- a/ML/2-modeling.R
+++ b/ML/2-modeling.R
@@ -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() %>%