ValueError:y_true中仅存在一个类。在这种情况下未定义ROC AUC分数

时间:2019-06-29 07:00:04

标签: python scikit-learn pytorch

我想使用所有实例为零的数据集来计算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

1 个答案:

答案 0 :(得分:0)

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