0

I am using the fit_resamples() function in TidyModels to get the F1 metrics as below. I would like to know how to pass the beta parameter whose default is set at 1 at the moment.

glm_workflow %>%
  fit_resamples(resamples = trainDatFolds,
                metrics = metric_set(roc_auc, pr_auc,
                                     accuracy, f_meas),
                control = control_resamples(save_pred = TRUE) %>%
  collect_metrics()

Thanks a lot!

Zarni

Zarni
  • 109
  • 6

1 Answers1

0

You'll need a simple wrapper for the metric. See ?metric_set. The examples include one where the ccc() function is used with an argument.

topepo
  • 13,534
  • 3
  • 39
  • 52