fig: summarize the full physical grid crossover

1 files changed, 6 insertions, 3 deletions

diff --git a/experiments/plot_physical_grid_p5.py b/experiments/plot_physical_grid_p5.py
index bc86647..35125d4 100644
--- a/experiments/plot_physical_grid_p5.py
+++ b/experiments/plot_physical_grid_p5.py
@@ -49,10 +49,10 @@ def parse_args() -> argparse.Namespace:
     parser = argparse.ArgumentParser()
     parser.add_argument(
         "--input", type=Path,
-        default=Path("results/physical_bias/p5_grid_bias_crossover.json"))
+        default=Path("results/physical_bias/p5_full_grid_bias_crossover.json"))
     parser.add_argument(
         "--output-prefix", type=Path,
-        default=Path("results/figs/physical_grid_p5"))
+        default=Path("results/figs/physical_grid_p5_full"))
     return parser.parse_args()
 
 
@@ -140,8 +140,11 @@ def main() -> None:
         axes[1], records, ("overclamp", "overclamp_sdil"),
         "(b) Composition with overclamping")
     axes[0].set_ylabel("Classification error (%)")
+    diameters = sorted({record["input_diameter_v"] for record in records})
+    trials_per_diameter = sum(
+        record["input_diameter_v"] == diameters[0] for record in records)
     axes[1].text(
-        0.98, 0.82, "8 trials per input diameter",
+        0.98, 0.62, f"{trials_per_diameter} trials per input diameter",
         transform=axes[1].transAxes, ha="right", va="top",
         fontsize=8, color="#666666")
     args.output_prefix.parent.mkdir(parents=True, exist_ok=True)