fixed min and max

JGIBristol · Nov 16, 2024 · e05c0da · e05c0da
1 parent 2b99e39
commit e05c0da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/ablate_attention.py b/scripts/ablate_attention.py
@@ -183,7 +183,7 @@ def main(args: argparse.Namespace):
     # Heatmap of Dice similarities
     fig, axis = plt.subplots()
 
-    im = axis.imshow(dice_matrix, cmap="plasma_r")
+    im = axis.imshow(dice_matrix, cmap="plasma_r", vmin=0, vmax=1)
     axis.set_xlabel("Removed attention block(s)")
     axis.set_ylabel("Removed attention block(s)")