我试图用下面的代码绘制一个混淆矩阵,但是我无法在每个框中显示数字!
这就是我所做的:
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()
答案 0 :(得分:0)
在kaggle上找到了answer。
您只需要插入此代码中的数字即可,例如:
from sklearn.metrics import confusion_matrix
cm = confusion_matrix (y_test, y_pred2)
print (cm)