experiment: implement KP mixed-traffic innovation controls

1 files changed, 8 insertions, 0 deletions

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())