confusion matrix fix

ragavvenkatesan · Mar 11, 2017 · 5b1d3d0 · 5b1d3d0
1 parent 82abc8e
commit 5b1d3d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yann/modules/resultor.py b/yann/modules/resultor.py
@@ -191,7 +191,7 @@ def _store_confusion_img (self, confusion, filename, verbose = 2):
         """
         if verbose >= 3:
             print ("... Saving the file down")
-        confusion = confusion / confusion.sum(axis = 0)
+        confusion = confusion / confusion.sum(axis = 1)
         fig = plt.figure()
         plt.matshow(confusion)
         plt.title('Confusion matrix')