ROC曲线报错bad input shape

最新推荐文章于 2022-04-17 20:23:29 发布

NewBeeFish

最新推荐文章于 2022-04-17 20:23:29 发布

阅读量1.2k

点赞数 1

文章标签：机器学习模型 ROC

本文链接：https://blog.csdn.net/weixin_41905712/article/details/102983516

版权

在进行二分类问题的机器学习模型评估时，尝试绘制ROC曲线时遇到了错误，提示输入形状不正确。原因是分类器预测输出的是样本在两类别的概率，而非单列的类别预测。为修复此问题，可以检查并确保输出是每个样本属于正类的概率。对于二分类，绘制ROC曲线正确的方法是使用每个样本对应的正类概率。

摘要由CSDN通过智能技术生成

ROC曲线报错

在使用分类器对二分类问题数据训练后，想要画ROC曲线，但出现如下报错
error: bad input shape (,2)

for k, (train_index, val_index) in enumerate(kf.split(x, y)):
    print('第%d个val' %(k+1))
    x_train = x.iloc[train_index,:]
    y_train = y.iloc[train_index,:]
    x_val = x.iloc[val_index,:]
    y_val = y.iloc[val_index,:]
    #ADASYN过采样
    o