From 1b72be4248e712d4808045cb9b999ade41a9ee41 Mon Sep 17 00:00:00 2001
From: Kyle Belanger <kyle.belanger@roche.com>
Date: Tue, 31 Jan 2023 16:04:06 -0500
Subject: [PATCH] Update 2-modeling.R

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

diff --git a/ML/2-modeling.R b/ML/2-modeling.R
index 93874f0..07b41cd 100644
--- a/ML/2-modeling.R
+++ b/ML/2-modeling.R
@@ -75,3 +75,10 @@ rf_predict <- ds_train %>%
 
 ys$accuracy(rf_predict, ft4_dia, .pred_class)
 ys$conf_mat(rf_predict, ft4_dia, .pred_class)
+
+ggplot2::autoplot(ys$roc_curve(rf_predict, ft4_dia, `.pred_Hypo`:`.pred_Non-Hyper`))
+
+ys$sensitivity(rf_predict,ft4_dia, .pred_class, estimator = "macro_weighted")
+ys$sensitivity(rf_predict,ft4_dia, .pred_class, estimator = "micro")
+ys$roc_auc(rf_predict, ft4_dia, `.pred_Hypo`, `.pred_Non-Hypo`, `.pred_Hyper`, `.pred_Non-Hyper`)
+