snapshot...

modules/util/plot_cm.py

@@ -59,7 +59,7 @@ def plot_confusion_matrix(cm, labels, title='Confusion matrix', filename = 'conf
     ax.yaxis.tick_left()
 
     for i, j in itertools.product(range(cm.shape[0]), range(cm.shape[1])):
-        ax.text(j, i, format(cm[i, j], 'd') if cm[i,j]!=0 else '.', horizontalalignment="center", fontsize=6, verticalalignment='center', color= "black")
+        ax.text(j, i, format(cm[i, j], 'd') if cm[i,j] != 0 else '.', horizontalalignment="center", fontsize=6, verticalalignment='center', color= "black")
     fig.set_tight_layout(True)
 
     ImageDirAndName = os.path.join('/Users/tomrink', filename)