sklearn:auc、roc_curve、roc_auc_score_NockinOnHeavensDoor的博客-CSDN博客
切记:
1。
train的时候,若出现valuerror,只出现一个class在y_true的情况时,可通过提高batchsize的大小解决,这种情况的发生是因为数据中一些数据是全1或者全0.
2。
再一个就是要明确函数里面的参数,是y_score,(根据模型的预测结果给的评分)而不是model的直接输出output。
正确:roc_auc_score(label,score)