Skip to content

Confusion Matrix and Classification Report

Yang YueXiang edited this page Sep 8, 2019 · 2 revisions

Refer to: 11.1 – 11.2 – MNIST Confusion Matrix Analysis and Viewing Misclassifications


from sklearn.metrics import classification_report,confusion_matrix
import numpy as np

y_pred = model.predict_classes(x_test)
#print(classification_report(y_test_classes, y_pred))
#print(confusion_matrix(y_test_classes, y_pred))
print(classification_report(np.argmax(y_test,axis=1), y_pred))
print(confusion_matrix(np.argmax(y_test,axis=1), y_pred))

Clone this wiki locally