如何在 TidyModel 中為 fit_resample 的 F1 分數添加 beta 參數

Question

我在 TidyModels 中使用 fit_resamples() function 來獲得如下 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() function 與參數一起使用的示例。