From c1ac6dd114abc7b7428c41ee3ac54ea7efd9417b Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Fri, 7 Aug 2026 16:02:49 -0500 Subject: fix: summarize Rain metric trajectories --- experiments/rain_ep_centered_r3.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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), -- cgit v1.2.3