diff --git a/R/compute_preds.R b/R/compute_preds.R new file mode 100644 index 0000000..09c81db --- /dev/null +++ b/R/compute_preds.R @@ -0,0 +1,14 @@ +# Function to perform tuning for each set of inner folds +compute_preds <- function(inner_resample, + xgb_mod, + xgb_recipe, + xgb_metrics, + outer_id) { + xgb_res <- tune_grid( + xgb_mod, + xgb_recipe, + resamples = inner_resample, + grid = xgb_grid, + metrics = xgb_metrics, + control = control_grid(save_pred = FALSE) + )