sklearn.confusion_matrix - TypeError：'numpy.ndarray'对象不可调用

Question

我正在尝试使用以下内容构建sklearn混淆矩阵

test_Y：

  Target
0   0
1   0
2   1

test_Y的数据类型是

Target int64
dtype: object

我的y_pred是

array([0,0,1])

然后我做我的混淆矩阵

cm = confusion_matrix(test_Y,y_pred)
sns.heatmap(cm,annot=True)

但我得到错误

TypeError：'numpy.ndarray'对象不可调用

Answer 1

你重复使用了confusion_matrix这个名字。你需要将它重新绑定回你的功能;这是一种方式：

from sklearn.metrics import confusion_matrix

cm = confusion_matrix(test_Y, y_pred)
sns.heatmap(cm, annot=True)