r - 是否有 R 函数来优化 PRG AUC（精确召回增益曲线下的面积）？

Question

为了优化 Caret 中的 PR AUC（精确召回曲线下的面积），可以将prSummary函数插入到trainControl函数中。

正如Flach 和 Kull介绍的那样，是否有类似的方法来优化 PRG AUC（精确召回增益曲线下的区域）？插入符号内部还是外部？

MLeval包返回 PRG 曲线和 PRG AUC，但似乎只在 Caret 训练输出上运行。

标签： rr-caretprecision-recall

Answer 1

事实证明，Meelis Kull为 PRG AUC 编写了一个 R 包。计算 PRG AUC 很简单：

prg_curve <- create_prg_curve(labels, scores)
prg_auc <- calc_auprg(prg_curve)

对于 Caret，这些函数可以插入到摘要函数中，正如上面评论中的链接中的 @missuse 所解释的那样。