diff --git a/utils/metrics.py b/utils/metrics.py index c01f823a7..7fb077774 100644 --- a/utils/metrics.py +++ b/utils/metrics.py @@ -208,7 +208,7 @@ class ConfusionMatrix: vmin=0.0, xticklabels=ticklabels, yticklabels=ticklabels).set_facecolor((1, 1, 1)) - ax.set_ylabel('True') + ax.set_xlabel('True') ax.set_ylabel('Predicted') ax.set_title('Confusion Matrix') fig.savefig(Path(save_dir) / 'confusion_matrix.png', dpi=250)