【发布时间】:2019-11-10 22:21:08
【问题描述】:
我想使用所有实例为零的数据集来计算 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
【问题讨论】:
标签: python machine-learning scikit-learn pytorch