Skip to content

Commit 6068c63

Browse files
andre15silvaclaude
andcommitted
fix(paper): remove shuffled series from combined layer AUC plots
The combined layer-AUC figures were showing shuffled run IDs as series labels instead of just the model lines + Random baseline. Now only the pooled model runs are plotted; the axhline at 0.5 serves as the baseline. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 783862f commit 6068c63

1 file changed

Lines changed: 5 additions & 10 deletions

File tree

run_paper_figures.py

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1118,20 +1118,15 @@ def main():
11181118
n_bins = args.n_bins, x_label = "Relative step bin", suffix = "_step",
11191119
)
11201120

1121-
# --- Layer AUC line plots (pooled runs only) ---
1121+
# --- Layer AUC line plots (pooled runs only, Random baseline from axhline) ---
11221122
print("[fig] AUC vs layer plots...")
1123-
pooled_ids = args.pooled_run_ids
1124-
all_run_ids = pooled_ids + (args.shuffled_run_ids or [])
1125-
all_labels = (
1126-
[MODEL_LABELS.get(r.replace("_pooled", ""), r) for r in pooled_ids]
1127-
+ [MODEL_LABELS.get(r.replace("_shuffled", ""), r) + " (shuffled)"
1128-
for r in (args.shuffled_run_ids or [])]
1129-
)
1123+
pooled_ids = args.pooled_run_ids
1124+
pool_labels = [MODEL_LABELS.get(_model_key(r), r) for r in pooled_ids]
11301125
for probe in probes:
11311126
plot_layer_auc(
11321127
results_dir = results_dir,
1133-
run_ids = all_run_ids,
1134-
run_labels = all_labels,
1128+
run_ids = pooled_ids,
1129+
run_labels = pool_labels,
11351130
probe = probe,
11361131
layers = layers,
11371132
figures_dir = figures_dir,

0 commit comments

Comments
 (0)