Do we have an ROC AUC Score as an Objective function in Pytorch?

1

Based on approximation of Wilcoxon-Mann-Whitney U statistic, here is the tflearn implementation

3 Likes
2

Any news on this? I think differentiable objective functions that directly optimize ROC-AUC and PRC-AUC scores will be useful in many scenarios.
There are some paper describing such functions:
ROC-AUC: Optimizing Classifier Performance via an Approximation to the Wilcoxon-Mann-Whitney Statistic (ICML 2003)
Paper (NIPS 2003)
PRC-AUC: [1608.04802] Scalable Learning of Non-Decomposable Objectives (ICML 2016)