python 计算roc程序报错ValueError: bad input shape (5313, 2)

返回概率的方法--predict_proba返回的是两列（我做的是二分类）

修改程序如下：

pred_proba = clf.predict_proba(test_x)[:, 1] # 本人将第二类为正样本，在此取了分类为第二类的的概率

参考链接：https://www.cnblogs.com/anovana/p/11750285.html