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

 

 

©️2020 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值