tidymodels - 如何在 TidyModel 中为 fit_resample 的 F1 分数添加 beta 参数

Question

我正在使用 TidyModels 中的 fit_resamples() 函数来获取 F1 指标，如下所示。我想知道如何传递默认设置为1的beta参数。

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

非常感谢！

扎尼

Answer 1

您需要一个简单的度量标准包装器。见?metric_set。这些示例包括将ccc()函数与参数一起使用的示例。