manual-sync 2026-07-02T21:01:54Z workspace/scripts

workspace/scripts/eval_positive_tangent_local_atlas.py

@@ -31,6 +31,8 @@ def main(argv: list[str] | None = None) -> int:
     parser.add_argument("--neighbor-pool", type=int, default=16)
     parser.add_argument("--utility-weight", type=float, default=0.0)
     parser.add_argument("--diversity-weight", type=float, default=0.0)
+    parser.add_argument("--negative-margin-weight", type=float, default=0.0)
+    parser.add_argument("--negative-pool", type=int, default=64)
     parser.add_argument("--all-tasks", action="store_true")
     parser.add_argument("--no-groups", action="store_true", help="Drop per-group rows from output.")
     args = parser.parse_args(argv)
@@ -44,6 +46,8 @@ def main(argv: list[str] | None = None) -> int:
         neighbor_pool=args.neighbor_pool,
         utility_weight=args.utility_weight,
         diversity_weight=args.diversity_weight,
+        negative_margin_weight=args.negative_margin_weight,
+        negative_pool=args.negative_pool,
         same_task_only=not args.all_tasks,
     )
     report = evaluate_local_atlas_retrieval(

workspace/scripts/summarize_positive_tangent_local_atlas_sweep.py

@@ -52,6 +52,8 @@ def _summarize(path: Path, data: dict[str, Any]) -> dict[str, Any]:
         "neighbor_pool": config.get("neighbor_pool"),
         "utility_weight": config.get("utility_weight"),
         "diversity_weight": config.get("diversity_weight"),
+        "negative_margin_weight": config.get("negative_margin_weight"),
+        "negative_pool": config.get("negative_pool"),
         "same_task_only": config.get("same_task_only"),
         "code_dim": data.get("code_dim"),
         "num_val_groups_with_positive": data.get("num_val_groups_with_positive"),
@@ -80,17 +82,19 @@ def _render_markdown(summary: dict[str, Any]) -> str:
     lines = [
         "# Positive Tangent Local-Atlas Retrieval Sweep",
         "",
-        "| rank | file | pool | utility | diversity | same task | K16 PTR@0.20 | K16 PTR@0.40 | K16 neg@0.20 | K16 pos<neg | mean pos dist |",
-        "|---:|---|---:|---:|---:|---|---:|---:|---:|---:|---:|",
+        "| rank | file | pool | utility | diversity | neg margin | neg pool | same task | K16 PTR@0.20 | K16 PTR@0.40 | K16 neg@0.20 | K16 pos<neg | mean pos dist |",
+        "|---:|---|---:|---:|---:|---:|---:|---|---:|---:|---:|---:|---:|",
     ]
     for index, row in enumerate(summary.get("rows", []), start=1):
         lines.append(
-            "| {rank} | {file} | {pool} | {utility} | {diversity} | {same_task} | {ptr02} | {ptr04} | {neg02} | {closer} | {dist} |".format(
+            "| {rank} | {file} | {pool} | {utility} | {diversity} | {neg_margin} | {neg_pool} | {same_task} | {ptr02} | {ptr04} | {neg02} | {closer} | {dist} |".format(
                 rank=index,
                 file=row["path"],
                 pool=row.get("neighbor_pool", "n/a"),
                 utility=_fmt(row.get("utility_weight")),
                 diversity=_fmt(row.get("diversity_weight")),
+                neg_margin=_fmt(row.get("negative_margin_weight")),
+                neg_pool=row.get("negative_pool", "n/a"),
                 same_task=str(row.get("same_task_only")),
                 ptr02=_pct(row.get("ptr_proxy_at_16_thr_0p2")),
                 ptr04=_pct(row.get("ptr_proxy_at_16_thr_0p4")),