用每个框中的数字制作一个混淆矩阵

时间:2018-07-12 01:09:20

标签: machine-learning confusion-matrix

我试图用下面的代码绘制一个混淆矩阵,但是我无法在每个框中显示数字!

这就是我所做的:

plt.matshow(cm, cmap= 'binary') #cubehelix, viridis, jet, PuOr, rainbow, RdBu
plt.title('Confusion matrix')
plt.colorbar()
plt.ylabel('True label')
plt.xlabel('Predicted label')
plt.show()

1 个答案:

答案 0 :(得分:0)

在kaggle上找到了answer

您只需要插入此代码中的数字即可,例如:

制作混淆矩阵来描述分类器的性能

from sklearn.metrics import confusion_matrix
cm = confusion_matrix (y_test, y_pred2)

print (cm)