saves the final results now

d4b58902 · Pamela Osuna · 2441b8bc · d4b58902
Commit d4b58902 authored Dec 10, 2019 by Pamela Osuna
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletions

confusion_matrix.py
+3 -1

No files found.
--- a/confusion_matrix.py
+++ b/confusion_matrix.py
@@ -87,4 +87,6 @@ def plot_confusion_matrix(cm,
    plt.tight_layout()
    plt.ylabel('True label')
    plt.xlabel('Predicted label\naccuracy={:0.4f}; misclass={:0.4f}'.format(accuracy, misclass))
-    plt.show()
+    #plt.show()
+    plt.savefig("/img/confusion_matrix")
+    plt.close()