from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay cm = confusion_matrix(y_test, y_pred) ConfusionMatrixDisplay(cm).plot() plt.savefig("confusion_matrix.png")