ml/gen: M1 数据生成器（采样/老师/距离下界认证/时间律/统计/一致性门禁）；ik-solver 增 clearance/gapMatrix/leverBounds

2026-10-17 15:00:09 +00:00
parent ea83de8bb0
commit c13366b830
10 changed files with 305 additions and 3 deletions
@@ -0,0 +1,39 @@
+/* M1 数据生成器（2026-10-17，方案 §4）
+ *   node ml/gen/gen_dataset.cjs --n 500 --seed 1 --out ml/data/shard_0001.jsonl
+ * 每行一条任务：采样 → 老师 → 距离下界认证过滤 → 局部时间律（50 Hz）→ 写出。
+ * 认证不过的样本也写出（keep=false + 原因），供 stats 统计丢弃率；训练只用 keep=true。 */
+'use strict';
+const fs = require('fs'), path = require('path');
+const args = {}; for (let i = 2; i < process.argv.length; i += 2) args[process.argv[i].replace(/^--/, '')] = process.argv[i + 1];
+const N = +args.n || 100, SEED = +args.seed || 1, OUT = args.out || 'ml/data/shard_' + String(SEED).padStart(4, '0') + '.jsonl';
+const { S, ArmIK } = require('../../tests/route_bench_setup.cjs')();
+const sampler = require('./sampler.cjs')(S, SEED * 7919 + 13);
+const teach = require('./teacher.cjs')(S, ArmIK);
+const LEVER = S.leverBounds(4000);
+const cert = require('./cert.cjs')(S, { lever: LEVER });
+const { timeParam } = require('./timelaw.cjs');
+const fp = S.modelFingerprint ? S.modelFingerprint() : null;
+fs.mkdirSync(path.dirname(OUT), { recursive: true });
+const fd = fs.openSync(OUT, 'w');
+const r3 = v => Math.round(v * 1000) / 1000;
+const t00 = performance.now();
+for (let i = 0; i < N; i++) {
+  const task = sampler.sample();
+  const t = teach(task);
+  const rec = { id: 's' + String(SEED).padStart(4, '0') + '_' + String(i).padStart(6, '0'), cat: task.cat, mode: task.mode,
+    q0: task.q0.map(r3), T: task.T.map(r3), R: task.R, fp };
+  if (!t.ok) { rec.keep = false; rec.why = t.why; fs.writeSync(fd, JSON.stringify(rec) + '\n'); continue; }
+  const c0 = performance.now();
+  const c = cert.path(t.pts);
+  Object.assign(rec, { path: t.pts.map(q => q.map(r3)), reached: t.reached, err: r3(t.err), oerr: r3(t.oerr), travel_deg: r3(t.travel),
+    plan_ms: Math.round(t.ms), blocked: t.blocked, poseRelaxed: t.poseRelaxed, detour: t.detour,
+    cert: { ok: c.ok, queries: c.queries, minGap: r3(c.minGap), ms: Math.round(performance.now() - c0), why: c.why || null, failAt: c.failAt == null ? null : c.failAt } });
+  /* 训练可用：认证通过；D 类允许未到达（停靠），其余类必须到达；poseRelaxed 的 pose 样本降为 pos */
+  rec.keep = c.ok && (t.reached || task.cat === 'D');
+  if (t.poseRelaxed && task.mode === 'pose') { rec.mode = 'pos'; rec.R = null; rec.demoted = true; rec.reached = t.err < 1; }
+  if (rec.keep) { const tp = timeParam(t.pts); rec.traj = tp.traj.map(q => q.map(r3)); rec.exec_s = r3(tp.steps * 0.02); }
+  fs.writeSync(fd, JSON.stringify(rec) + '\n');
+  if ((i + 1) % 25 === 0) process.stderr.write(`[seed ${SEED}] ${i + 1}/${N}  ${((performance.now() - t00) / 1000 / (i + 1)).toFixed(2)} s/条\n`);
+}
+fs.closeSync(fd);
+process.stderr.write(`[seed ${SEED}] done ${N} in ${((performance.now() - t00) / 1000).toFixed(0)} s\n`);