From 207616a91c7a1e40105d7127a0d2b22651bcce19 Mon Sep 17 00:00:00 2001
From: Kyle Belanger <kyleb44@hotmail.com>
Date: Thu, 2 Feb 2023 20:50:30 -0500
Subject: [PATCH] Update 2-modeling.R

---
 ML/2-modeling.R | 28 ++++++++++++++++++++++++----
 1 file changed, 24 insertions(+), 4 deletions(-)

diff --git a/ML/2-modeling.R b/ML/2-modeling.R
index 1245e7e..f2ed7e1 100644
--- a/ML/2-modeling.R
+++ b/ML/2-modeling.R
@@ -92,9 +92,29 @@ data_fold <- rsamp$vfold_cv(class_train, v = 5)
 
 rf_workflow <- wf$update_model(rf_workflow, rf_tuning_model)
 
-rf_tune <- rf_workflow %>%
-    tune::tune_grid(
-    data_fold
-    ,grid = rf_param %>% d$grid_regular()
+# takes around 1 hr to run grid search.  saving best params manaually
+# rf_tune <- rf_workflow %>%
+#     tune::tune_grid(
+#     data_fold
+#     ,grid = rf_param %>% d$grid_regular()
+#   )
+
+rf_best_params <- tibble::tibble(
+  mtry = 8
+  ,trees = 2000
+  ,min_n = 2
+)
+
+final_rf_workflow <- rf_workflow %>%
+  tune::finalize_workflow(rf_best_params)
+
+final_rf_fit <- p$fit(final_rf_workflow, class_train)
+
+final_rf_predict <- class_train %>%
+  dplyr::select(ft4_dia) %>%
+  dplyr::bind_cols(
+    predict(final_rf_fit, class_train)
+    ,predict(final_rf_fit, class_train, type = "prob")
   )
 
+final_conf_rf <- ys$conf_mat(final_rf_predict, ft4_dia, .pred_class)