使用 AUC 指标进行多标签分类的预测

Predictions using AUC metrics for multilabel classification

我正在使用 AUC 指标进行多标签分类。由于keras移除了prediction_classes以获得预测类，我只是使用0.5的阈值来获得输出类。但是，据我了解，对于不平衡数据集，AUC 的阈值不应为 0.5。如何获得用于训练模型的阈值？

另外，我知道AUC是用来做二分类的。我可以只将它用于多标签问题吗？如何计算阈值？取平均值与否。

多标签问题可以使用AUC，勾选this。

import numpy as np 
y_true = np.random.randint(0,2,(100,4))
y_pred = np.random.randint(0,2,(100,4))

m = tf.keras.metrics.AUC(multi_label=True, thresholds=[0, 0.5])
m(y_true, y_pred).numpy()

仅供参考，从 tf 2.5 开始，它现在支持对数预测。