Hark
5591adb39a
- B-1 运行时投影护栏恢复 5 mm；老师轨迹按认证最小间隙 < 3 mm 剔除（stats/build_dataset，gapDrop） - B-2 APG 端点+中点 hinge 取均值；B-3 check_r2 空匹配保护、5000×50000、分块 256，pass=false 时 prep 停下 - B-4 R7 按全部 B/C 题统计；B-5 r8 起点贴柱；B-7 pipeline r4（export_onnx --random，训练前测延迟） - B-8 起步失败 ≤0.2%、column_gap 分块、pipefail、门禁结果先写 .tmp - 训练控制台：ml/console.bat|sh → ml/monitor.py（开始/停止/进度/日志接口）+ monitor.html； pipeline.sh all 支持 FROM/TO，阶段状态写 ml/runs/pipeline.json（ml/pstate.py）；训练/APG 日志加 total、t - 文档：README 一键运行与训练控制台；审查文档追加第三轮处置；方案余量决策与 R4 时点
50 lines
3.1 KiB
Python
50 lines
3.1 KiB
Python
"""方案 R2 数据侧检查：近邻样本的标签是否自洽（时间律不可观测成分会让相同输入对应不同动作）。
|
||
python -m ml.eval.check_r2 --samples ml/data/samples --out ml/data/r2.json
|
||
做法：特征逐维标准化；随机取 Q 个查询样本，在 P 个样本池里找最近邻（排除同一任务，避免同轨迹相邻帧冒充"近邻"）；
|
||
取距离最近的 5% 近邻对，比较前 3 步（实际提交的动作）标签：std ≈ RMS(ΔY)/√2，单位为每步最大步长 STEP 的比例。
|
||
判据：label_std < 0.10（即 < 10% 步长）。起步失败率在闭环仿真结果的 start_fail 字段（acceptance 汇总）。
|
||
2026-10-18 新增（审查：R2 无执行点）。默认 5000 查询 × 50000 池，分块 256：GPU 峰值 < 0.5 GB，CPU 约 1–2 分钟；近邻对为空时输出 pass=null 而不报错。"""
|
||
import argparse, json, math
|
||
import torch
|
||
|
||
from ml.data import Samples
|
||
|
||
|
||
def main():
|
||
ap = argparse.ArgumentParser()
|
||
ap.add_argument('--samples', nargs='+', default=['ml/data/samples']); ap.add_argument('--out', default='ml/data/r2.json')
|
||
ap.add_argument('--queries', type=int, default=5000); ap.add_argument('--pool', type=int, default=50000)
|
||
ap.add_argument('--frac', type=float, default=0.05); ap.add_argument('--seed', type=int, default=0)
|
||
ap.add_argument('--device', default='cuda' if torch.cuda.is_available() else 'cpu')
|
||
a = ap.parse_args()
|
||
d = Samples(a.samples, device=a.device, split='train')
|
||
g = torch.Generator(device=a.device); g.manual_seed(a.seed)
|
||
P = min(a.pool, d.n); Q = min(a.queries, d.n)
|
||
pi = torch.randperm(d.n, device=a.device, generator=g)[:P]; qi = torch.randperm(d.n, device=a.device, generator=g)[:Q]
|
||
X = d.X.float(); mu, sd = X[pi].mean(0), X[pi].std(0).clamp(min=1e-6)
|
||
Xp, Xq = (X[pi] - mu) / sd, (X[qi] - mu) / sd
|
||
tp, tq = d.TID[pi], d.TID[qi]
|
||
best_d = torch.empty(Q, device=a.device); best_j = torch.empty(Q, dtype=torch.long, device=a.device)
|
||
for s in range(0, Q, 256):
|
||
D = torch.cdist(Xq[s:s + 256], Xp)
|
||
D.masked_fill_(tq[s:s + 256, None] == tp[None], float('inf'))
|
||
v, j = D.min(1); best_d[s:s + 256] = v; best_j[s:s + 256] = j
|
||
fin = torch.isfinite(best_d)
|
||
if fin.sum() < 20:
|
||
out = {'pairs': 0, 'label_std_mean': None, 'pass': None, 'rule': 'label_std_mean < 0.10', 'note': '近邻对不足（样本太少），未判定'}
|
||
print(json.dumps(out)); json.dump(out, open(a.out, 'w'), indent=1); return
|
||
thr = torch.quantile(best_d[fin], a.frac)
|
||
m = fin & (best_d <= thr)
|
||
Yq, Yp = d.Y[qi][m, :3].float(), d.Y[pi][best_j[m], :3].float()
|
||
rms = (Yq - Yp).pow(2).mean(-1).mean(-1).sqrt() # 每对、6 轴 × 3 步平均
|
||
std = (rms / math.sqrt(2))
|
||
out = {'pairs': int(m.sum()), 'nn_dist_threshold': thr.item(), 'label_std_mean': std.mean().item(),
|
||
'label_std_p90': torch.quantile(std, 0.9).item(), 'pass': std.mean().item() < 0.10,
|
||
'rule': 'label_std_mean < 0.10（10% 步长）'}
|
||
print(json.dumps(out, indent=1))
|
||
json.dump(out, open(a.out, 'w'), indent=1)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|