valueerror:y_true中只有一个类.在这种情况下,ROC AUC得分未定义



我想使用所有实例零的数据集来计算AUROC。但是,发生以下错误:

valueerror:y_true中只有一个类。在这种情况下,ROC AUC得分未定义

def computeAUROC (dataGT, dataPRED, classCount):
    outAUROC = []
    datanpGT = dataGT.cpu().numpy()
    datanpPRED = dataPRED.cpu().numpy()
    for i in range(classCount):
        outAUROC.append(roc_auc_score(datanpGT[:, i], datanpPRED[:, i]))
    return outAUROC

如果没有数据集中的正面和负面示例,就无法拥有ROC曲线。数据集中只有一个类,您无法测量假阳性率,因此无法绘制ROC曲线。这就是为什么您会收到此错误消息。

最新更新