如何计算整个时代的 roc auc 分数，如平均准确度？

【问题标题】：How to calculate roc auc score for the whole epoch like avg accuracy?如何计算整个时代的 roc auc 分数，如平均准确度？
【发布时间】：2020-05-13 06:56:46
【问题描述】：

我正在 PyTorch 中实现一个训练循环，对于指标，我想通过sklearn.metrics.roc_auc_score 使用 ROC AUC 分数。

我可以使用 sklearn 的实现来计算单个预测的分数，但是很难想象如何使用它来计算整个时期的平均分数。谁能把我推向正确的方向？

【问题讨论】：

标签： python scikit-learn pytorch metrics

【解决方案1】：

函数中的y_true和y_score,可以是一维数组，所以如果你收集整个epoch的值，你可以直接调用函数。请注意，如果您进行多标签分类，则需要分别计算每个类别的 ROC AUC 分数。

【讨论】：