added figure suptitles

LostOxygen · Jul 10, 2023 · 857261c · 857261c
1 parent 6bff4d7
commit 857261c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/visualize_lrp.py b/visualize_lrp.py
@@ -176,6 +176,7 @@ def main() -> None:
 
         # create and save the normal LRP analysis
         fig, axes = plt.subplots(1, 3, figsize=(10, 3))
+        fig.suptitle(f"LRP Analysis - {model_type}")
         plt.colorbar(mpl.cm.ScalarMappable(norm=norm, cmap=cmap),
                      orientation="vertical", ax=axes,
                      label="Attribution Values")
@@ -200,6 +201,7 @@ def main() -> None:
 
         # create and save the LRP analysis for the specral channels
         fig, axes = plt.subplots(1, 4, figsize=(20, 5))
+        fig.suptitle(f"Spectral LRP Analysis - {model_type}")
         axes[0].grid()
         axes[0].imshow(img)
         axes[0].set_title("Input Image")