From 4974a4c86cb383261a35b287acc3accd7482460c Mon Sep 17 00:00:00 2001
From: Kyle Belanger <kyleb44@hotmail.com>
Date: Fri, 3 Feb 2023 14:37:02 -0500
Subject: [PATCH] Update 2-modeling.R

---
 ML/2-modeling.R | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/ML/2-modeling.R b/ML/2-modeling.R
index f2ed7e1..07c5f25 100644
--- a/ML/2-modeling.R
+++ b/ML/2-modeling.R
@@ -118,3 +118,29 @@ final_rf_predict <- class_train %>%
   )
 
 final_conf_rf <- ys$conf_mat(final_rf_predict, ft4_dia, .pred_class)
+
+
+# random forest regression ------------------------------------------------
+
+reg_metrics <- ys$metric_set(ys$rmse, ys$rsq, ys$mae)
+
+rf_base_reg_model <- p$rand_forest() %>%
+  p$set_engine("ranger") %>% p$set_mode("regression")
+
+rf_reg_recipe <- r$recipe(FT4 ~ . , data = reg_train) %>%
+  r$update_role(subject_id, new_role = "id") %>%
+  r$update_role(charttime, new_role = "time") %>%
+  r$step_impute_bag(r$all_predictors())
+
+
+rf_reg_workflow <- wf$workflow() %>%
+  wf$add_model(rf_base_reg_model) %>%
+  wf$add_recipe(rf_reg_recipe)
+
+rf_base_reg_fit <- p$fit(rf_reg_workflow, reg_train)
+
+rf_reg_predict <- reg_train %>%
+  dplyr::select(FT4) %>%
+  dplyr::bind_cols(
+    predict(rf_base_reg_fit, reg_train)
+  )