我尝试使用下面的代码绘制混淆矩阵,但无法在每个方框中显示数字!
这是我所做的:
plt.matshow(cm, cmap= 'binary') #cubehelix, viridis, jet, PuOr, rainbow, RdBuplt.title('Confusion matrix')plt.colorbar()plt.ylabel('True label')plt.xlabel('Predicted label')plt.show()
回答:
在Kaggle上找到了答案。
你只需要将你的数字插入到这个代码示例中即可:
制作混淆矩阵以描述分类器的性能
from sklearn.metrics import confusion_matrixcm = confusion_matrix (y_test, y_pred2)print (cm)