diff options
Diffstat (limited to 'experiments/rain_ep_centered_r3.sh')
| -rwxr-xr-x | experiments/rain_ep_centered_r3.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/experiments/rain_ep_centered_r3.sh b/experiments/rain_ep_centered_r3.sh index 2ea5acb..be44649 100755 --- a/experiments/rain_ep_centered_r3.sh +++ b/experiments/rain_ep_centered_r3.sh @@ -54,7 +54,7 @@ by_seed = {} for path in sorted(root.glob("seed_*.json")): _, seed, method = path.stem.split("_", 2) report = json.loads(path.read_text()) - trajectory = [epoch["test_accuracy"] for epoch in report["epochs"]] + trajectory = [epoch["test_accuracy"] for epoch in report["metrics"]] by_seed.setdefault(seed, {})[method] = { "trajectory": trajectory, "best_accuracy": max(trajectory), |