kyle.belanger/DHSC-Capstone
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Update 2-modeling.R

Browse source
This commit is contained in:
Kyle Belanger 2023-02-02 20:50:30 -05:00
parent 9cb7fe3e13
commit 207616a91c
1 changed files with 24 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
ML/2-modeling.R
View file

@ -92,9 +92,29 @@ data_fold <- rsamp$vfold_cv(class_train, v = 5)
rf_workflow <- wf$update_model(rf_workflow, rf_tuning_model) rf_workflow <- wf$update_model(rf_workflow, rf_tuning_model)
rf_tune <- rf_workflow %>% # takes around 1 hr to run grid search. saving best params manaually
tune::tune_grid( # rf_tune <- rf_workflow %>%
data_fold # tune::tune_grid(
,grid = rf_param %>% d$grid_regular() # 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)