【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类型即可解决。

 

 

发布了392 篇原创文章 · 获赞 487 · 访问量 240万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览