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

Update 2-modeling-reg.R

Browse source
This commit is contained in:
Kyle Belanger 2023-03-20 14:54:03 -04:00
parent f504a52bb5
commit b51e1156d3
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -124,7 +124,7 @@ normalized <-
workflowsets::workflow_set(
preproc = list(normalized = normalized_rec),
models = list(
# KNN = knn_spec,
KNN = knn_spec,
neural_network = nnet_spec)
) %>%
workflowsets::option_add(param_info = nnet_param, id = "normalized_neural_network")
@ -145,13 +145,19 @@ all_workflows <-
# grid search -------------------------------------------------------------
num_cores <- parallel::detectCores() - 1
grid_ctrl <-
tune$control_grid(
save_pred = TRUE,
parallel_over = "everything",
save_workflow = TRUE
save_workflow = TRUE,
verbose = TRUE
)
doParallel::registerDoParallel(cores = num_cores)
grid_results <-
all_workflows %>%
workflowsets::workflow_map(