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)