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)