【问题标题】:Is there an R function to optimize the PRG AUC (area under the precision-recall-gain curve)?是否有 R 函数来优化 PRG AUC(精确召回增益曲线下的面积)?
【发布时间】:2020-07-01 09:16:54
【问题描述】:

要优化 Caret 中的 PR AUC(精确召回曲线下的面积),可以将 prSummary 函数插入到 trainControl 函数中。

有没有类似Flach and Kull 介绍的方法来优化 PRG AUC(precision-recall-gain 曲线下的区域)? Caret 内部还是外部?

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

【问题讨论】:

标签: r r-caret precision-recall


【解决方案1】:

事实证明,Meelis Kull 写了一封 R package for the PRG AUC。计算 PRG AUC 很简单:

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

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

【讨论】:

    猜你喜欢
    • 2012-09-03
    • 2016-06-11
    • 1970-01-01
    • 2014-09-21
    • 2018-09-12
    • 2017-09-21
    • 2020-04-18
    • 2021-09-13
    • 2019-06-22
    相关资源
    最近更新 更多