From a4858a04acc5adca65a437295b63b426b6358831 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Wed, 22 Jul 2026 14:46:41 -0500 Subject: experiment: implement KP mixed-traffic innovation controls --- experiments/analyze_kp_innovation_short.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'experiments/analyze_kp_innovation_short.py') diff --git a/experiments/analyze_kp_innovation_short.py b/experiments/analyze_kp_innovation_short.py index 4c70575..5f21dea 100644 --- a/experiments/analyze_kp_innovation_short.py +++ b/experiments/analyze_kp_innovation_short.py @@ -82,12 +82,20 @@ def main(): if len(tracking[rule]) != 20 or any(value is None for value in tracking[rule]): raise ValueError(f"MT-1 {rule} tracking trajectory is incomplete") for row, values in zip(records[rule]["epochs"], tracking[rule]): + mixed = row.get("mixed_apical") + if mixed is None: + raise ValueError(f"MT-1 {rule} mixed-apical trajectory is incomplete") trajectory_values.extend([ float(row["train_loss"]), float(values["mean_feedback_forward_cosine"]), float(values["mean_feedback_forward_relative_error"]), float(values["min_feedback_forward_cosine"]), float(values["max_feedback_forward_relative_error"]), + float(mixed["teaching_rms"]), + float(mixed["instruction_rms"]), + float(mixed["raw_apical_rms"]), + float(mixed["innovation_rms"]), + float(mixed["traffic_rms"]), ]) all_finite = all(record["final"]["finite"] for record in records.values()) -- cgit v1.2.3