diff --git a/modules/util/plot_cm.py b/modules/util/plot_cm.py index 619d00cdf025ac9da2a6e6fb8365ad8984e112f2..bca19b5bea79d846aa45ad4118451a1e052006fa 100644 --- a/modules/util/plot_cm.py +++ b/modules/util/plot_cm.py @@ -65,5 +65,7 @@ def plot_confusion_matrix(cm, labels, title='Confusion matrix', filename = 'conf 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) + plt.title(title, loc='left', fontweight='bold', fontsize=6) + ImageDirAndName = os.path.join('/Users/tomrink', filename) fig.savefig(ImageDirAndName)