diff options
Diffstat (limited to 'experiments')
| -rw-r--r-- | experiments/plot_physical_grid_p5.py | 9 |
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) |