From d01c4a3193387c340c388af46d7399bfadbb7eab Mon Sep 17 00:00:00 2001 From: lrusso96 Date: Sun, 9 Dec 2018 19:11:53 +0100 Subject: [PATCH] fix plot of confusion matrix --- inceptionv3_svm_classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inceptionv3_svm_classifier.py b/inceptionv3_svm_classifier.py index 4358161..c9bc3d5 100644 --- a/inceptionv3_svm_classifier.py +++ b/inceptionv3_svm_classifier.py @@ -97,7 +97,7 @@ def plot_confusion_matrix(y_true, y_pred, matrix_title): # use sklearn confusion matrix cm_array = confusion_matrix(y_true, y_pred) - plt.imshow(cm_array[:-1, :-1], interpolation='nearest', cmap=plt.cm.Blues) + plt.imshow(cm_array, interpolation='nearest', cmap=plt.cm.Blues) plt.title(matrix_title, fontsize=16) cbar = plt.colorbar(fraction=0.046, pad=0.04)