From 12dad3bc87bc1ff059d0a2d390f2ff0829650e0d Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 21 Oct 2022 15:22:35 +0100 Subject: [PATCH] vis/segmentation: fix titles --- aimodel/src/lib/vis/segmentation_plot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/vis/segmentation_plot.py b/aimodel/src/lib/vis/segmentation_plot.py index 4e4901c..a018f03 100644 --- a/aimodel/src/lib/vis/segmentation_plot.py +++ b/aimodel/src/lib/vis/segmentation_plot.py @@ -21,7 +21,7 @@ def segmentation_plot(water_actual, water_predict, model_code, filepath_output): axes["B"].imshow(water_predict) - axes["A"].set_title(f"Predicted", fontsize=20) + axes["B"].set_title(f"Predicted", fontsize=20) plt.suptitle(f"Rainfall → Water depth prediction | {model_code}", fontsize=28, weight="bold")