Add files via upload

visualize.py

@@ -0,0 +1,114 @@
+#!/usr/bin/env python3
+"""
+Visualize RYS sweep results.
+Reads the JSONL output from sweep.py, prints ranked table and bar chart.
+"""
+
+import json
+import sys
+
+
+def load_results(path: str):
+    baseline = None
+    results = []
+    with open(path) as f:
+        for line in f:
+            line = line.strip()
+            if not line:
+                continue
+            entry = json.loads(line)
+            if entry.get("is_baseline"):
+                baseline = entry
+            else:
+                results.append(entry)
+    return baseline, results
+
+
+def print_ranked(baseline, results):
+    """Print results ranked by combined delta."""
+    if not baseline:
+        print("No baseline found in results!")
+        return
+
+    for r in results:
+        math_delta = r["math_score"] - baseline["math_score"]
+        eq_delta = r["eq_score"] - baseline["eq_score"]
+        r["math_delta"] = math_delta
+        r["eq_delta"] = eq_delta
+        r["combined"] = (math_delta * 100) + eq_delta
+
+    ranked = sorted(results, key=lambda x: x["combined"], reverse=True)
+
+    print(f"\nBaseline: math={baseline['math_score']:.4f}  eq={baseline['eq_score']:.2f}")
+    print()
+    print(f"{'Rank':>4} {'Config':>12} {'Dup':>4} "
+          f"{'Math':>8} {'EQ':>8} "
+          f"{'Math Δ':>9} {'EQ Δ':>8} {'Combined':>10}")
+    print("-" * 80)
+
+    for i, r in enumerate(ranked):
+        config = f"({r['dup_start']},{r['dup_end']})"
+        n_dup = r['dup_end'] - r['dup_start']
+
+        if r["combined"] > 0:
+            marker = "+"
+        elif r["combined"] < -5:
+            marker = "!"
+        else:
+            marker = " "
+
+        print(f"{i+1:>4} {config:>12} {n_dup:>4} "
+              f"{r['math_score']:>8.4f} {r['eq_score']:>8.2f} "
+              f"{r['math_delta']:>+9.4f} {r['eq_delta']:>+8.2f} "
+              f"{r['combined']:>+10.2f} {marker}")
+
+    if ranked:
+        best = ranked[0]
+        worst = ranked[-1]
+        print()
+        print(f"Best:  ({best['dup_start']},{best['dup_end']})  combined={best['combined']:+.2f}")
+        print(f"Worst: ({worst['dup_start']},{worst['dup_end']})  combined={worst['combined']:+.2f}")
+
+
+def print_bar_chart(baseline, results):
+    """Print a horizontal bar chart sorted by start position."""
+    if not baseline or not results:
+        return
+
+    for r in results:
+        math_delta = r["math_score"] - baseline["math_score"]
+        eq_delta = r["eq_score"] - baseline["eq_score"]
+        r["combined"] = (math_delta * 100) + eq_delta
+
+    ordered = sorted(results, key=lambda x: x["dup_start"])
+
+    max_val = max(abs(r["combined"]) for r in ordered)
+    if max_val == 0:
+        max_val = 1
+
+    half = 20
+    print(f"\nCombined delta (baseline = |):")
+    print(f"{'Config':>12} {'negative':<{half}}|{'positive':<{half}}")
+
+    for r in ordered:
+        config = f"({r['dup_start']},{r['dup_end']})"
+        val = r["combined"]
+        bar_len = int(abs(val) / max_val * half)
+
+        if val >= 0:
+            bar = " " * half + "|" + "#" * bar_len
+        else:
+            pad = half - bar_len
+            bar = " " * pad + "=" * bar_len + "|"
+
+        print(f"{config:>12} {bar} {val:+.2f}")
+
+
+if __name__ == "__main__":
+    if len(sys.argv) < 2:
+        print(f"Usage: {sys.argv[0]} <results.jsonl>")
+        sys.exit(1)
+
+    baseline, results = load_results(sys.argv[1])
+    print_ranked(baseline, results)
+    print_bar_chart(baseline, results)