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( workflowsets::workflow_set(
preproc = list(normalized = normalized_rec), preproc = list(normalized = normalized_rec),
models = list( models = list(
# 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")
@ -145,13 +145,19 @@ all_workflows <-
# grid search ------------------------------------------------------------- # grid search -------------------------------------------------------------
num_cores <- parallel::detectCores() - 1
grid_ctrl <- grid_ctrl <-
tune$control_grid( tune$control_grid(
save_pred = TRUE, save_pred = TRUE,
parallel_over = "everything", parallel_over = "everything",
save_workflow = TRUE save_workflow = TRUE,
verbose = TRUE
) )
doParallel::registerDoParallel(cores = num_cores)
grid_results <- grid_results <-
all_workflows %>% all_workflows %>%
workflowsets::workflow_map( workflowsets::workflow_map(