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

updates and ML output

Browse source 
updates
This commit is contained in:
Kyle Belanger 2023-03-22 12:06:45 -04:00
parent fdc71216dd
commit b76e141aee
1 changed files with 14 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
ML/2-modeling-reg.R
View file

@ -71,7 +71,9 @@ normalized_rec <- recipes::recipe(FT4 ~ ., data = ds_train) %>%
rf_rec <- recipes::recipe(FT4 ~ . , data = ds_train) %>% rf_rec <- recipes::recipe(FT4 ~ . , data = ds_train) %>%
recipes::step_impute_bag(recipes::all_predictors()) recipes::step_impute_bag(recipes::all_predictors())
boost_rec <- recipes::recipe(FT4 ~ . , data = ds_train) %>%
recipes::step_impute_bag(recipes::all_predictors()) %>%
recipes::step_dummy(gender)
# models ------------------------------------------------------------------ # models ------------------------------------------------------------------
@ -134,29 +136,34 @@ normalized <-
workflowsets::workflow_set( workflowsets::workflow_set(
preproc = list(normalized = normalized_rec), preproc = list(normalized = normalized_rec),
models = list( models = list(
# SVM_radial = svm_r_spec, SVM_radial = svm_r_spec,
# SVM_poly = svm_p_spec, # SVM_poly = svm_p_spec,
KNN = knn_spec, KNN = knn_spec,
neural_network = nnet_spec) neural_network = nnet_spec)
) %>% ) %>%
workflowsets::option_add(param_info = nnet_param, id = "normalized_neural_network") workflowsets::option_add(param_info = nnet_param, id = "normalized_neural_network")
forests <- forest <-
workflowsets::workflow_set( workflowsets::workflow_set(
preproc = list(forests = rf_rec), preproc = list(forests = rf_rec),
models = list(RF = rf_spec, boosting = xgb_spec) models = list(RF = rf_spec)
) %>% ) %>%
workflowsets::option_add(param_info = rf_param, id = "forests_RF") workflowsets::option_add(param_info = rf_param, id = "forests_RF")
boost <-
workflowsets::workflow_set(
preproc = list(boost = boost_rec),
models = list(boosting = xgb_spec)
)
all_workflows <- all_workflows <-
dplyr::bind_rows(normalized, forests) %>% dplyr::bind_rows(normalized, forest, boost) %>%
dplyr::mutate(wflow_id = gsub("(forests_)|(normalized_)", "", wflow_id)) dplyr::mutate(wflow_id = gsub("(forest_)|(normalized_)|(boost_)", "", wflow_id))
# workflow screening ------------------------------------------------------ # workflow screening ------------------------------------------------------
num_cores <- parallel::detectCores() - 1 num_cores <- parallel::detectCores() - 2
doParallel::registerDoParallel(cores = num_cores) doParallel::registerDoParallel(cores = num_cores)
screen_workflows <- all_workflows %>% screen_workflows <- all_workflows %>%