26 lines
1.4 KiB
JavaScript
26 lines
1.4 KiB
JavaScript
|
/* 对拍数据：JS 侧 fk / orient / gapMatrix / 特征（ik-policy.js）→ ml/data/parity.json
|
||
|
|
* node ml/eval/dump_parity.cjs [N=2000]
|
||
|
|
* 然后 python -m ml.eval.check_parity 比对 PyTorch 版。 */
|
||
|
|
'use strict';
|
||
|
|
const fs = require('fs'), path = require('path');
|
||
|
|
const N = +process.argv[2] || 2000;
|
||
|
|
const { S } = require('../../tests/route_bench_setup.cjs')();
|
||
|
|
const POL = require('../../webapp/js/ik-policy.js');
|
||
|
|
const L = S.LIM, O = S.ORDER;
|
||
|
|
let st = 99; const rnd = () => { st = (st * 1664525 + 1013904223) >>> 0; return st / 4294967296; };
|
||
|
|
const rq = () => O.map(id => L[id].lo + rnd() * (L[id].hi - L[id].lo));
|
||
|
|
const recs = [];
|
||
|
|
for (let n = 0; n < N; n++) {
|
||
|
|
const q = rq(), qp = q.map((v, i) => v - (rnd() - 0.5) * 2);
|
||
|
|
const b = rq(); const T = S.fk(b, 0); const pose = rnd() < 0.5; const R = pose ? S.orient(b) : null;
|
||
|
|
const r = Math.floor(rnd() * 4); const prefix = []; for (let k = 0; k < r; k++) prefix.push(q.map(() => (rnd() - 0.5) * 2));
|
||
|
|
const sub = rnd() < 0.5 ? rq() : null;
|
||
|
|
const st0 = { q, qPrev: qp, T, R, prefix, sub };
|
||
|
|
const f = POL.features(S, st0);
|
||
|
|
recs.push({ q, qPrev: qp, T, R, prefix, sub, fk: S.fk(q, 0), orient: S.orient(q), gm: Array.from(S.gapMatrix(q, 40)), feat: Array.from(f) });
|
||
|
|
}
|
||
|
|
const out = path.join(__dirname, '../data/parity.json');
|
||
|
|
fs.mkdirSync(path.dirname(out), { recursive: true });
|
||
|
|
fs.writeFileSync(out, JSON.stringify({ fp: S.modelFingerprint(), recs }));
|
||
|
|
console.log('wrote', out, recs.length);
|