diff --git a/modules/util/plot_cm.py b/modules/util/plot_cm.py index 4638ab8998496fdc3504234b79491749ee82a749..67f09d1ba8727790c5a95cfd0e3878fb73037534 100644 --- a/modules/util/plot_cm.py +++ b/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)