【Python】获取roc、auc时候报错：raise ValueError("{0} format is not supported".format(y_type))

代码：

fpr, tpr, thresholds = roc_curve(y_test, test_prob, pos_label=1, sample_weight=None, drop_intermediate=True)

报错：

raise ValueError("{0} format is not supported".format(y_type))
ValueError: unknown format is not supported

 

原因：roc_curve中的y_true不符合要求，只支持两种情况：binary、multil-class & pos_label = 1。可以用如下函数检查

from sklearn.utils.multiclass import type_of_target
print(type_of_target(y_test))

如果结果是unknown，则说明类型无法识别。 把y_test 转化为list、narray类型即可解决。