Commit Graph

27 Commits

Author SHA1 Message Date
d1ecb13175 refactor: move server/ to tools/server/ 2026-05-12 00:37:47 +08:00
e0b3b012e7 refactor: move collector to tools/ 2026-05-12 00:35:46 +08:00
6c96ab68c8 refactor(scroll): move trainer/models/collector to tools/scroll/ 2026-05-12 00:34:05 +08:00
ba52c49edf refactor: move trainer/models/utils/config to tools/ 2026-05-12 00:30:55 +08:00
c89025047c chore: scaffold tools/ directory 2026-05-12 00:23:03 +08:00
e3d4626031 test: capture scroll generate() golden output (pre-migration) 2026-05-12 00:20:13 +08:00
98daef54ca test: capture mouse generate() golden output (pre-migration) 2026-05-12 00:19:39 +08:00
99f7ae29cb chore(scroll): train initial scroll model from scroll_traces.jsonl 2026-05-12 00:17:12 +08:00
8003ae657a docs: implementation plan for ai_mouse library refactor
Expands the 2026-05-11 design spec into ~40 bite-sized tasks across 6
phases (pre-flight golden capture, tools/ extraction, src layout switch,
ONNX export, NumPy/ORT rewrite, docs cleanup). Each task is self-contained
with full code blocks, exact file paths, and verification commands. TDD
where applicable; pure-move tasks use shorter scaffolding.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-05-12 00:01:37 +08:00
74d6c5c7ff docs: design spec for ai_mouse library refactor (ONNX + src-layout)
Captures the decisions made in brainstorming for 0.2.0:
- Split src/ai_mouse/ (pure-inference, numpy+onnxruntime only) from
  tools/ (training/server/eval, torch+fastapi+...)
- Bundled ONNX weights via importlib.resources
- Public API: MouseModel/ScrollModel classes + cached generate() helpers
- ONNX export script with PyTorch parity check
- Golden tests to lock semantics during NumPy rewrite of post-processing
- 5-stage migration plan, git URL install, no PyPI

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-05-11 23:40:21 +08:00
8b0447b949 docs: add CLAUDE.md guide for future Claude Code sessions 2026-05-10 13:54:12 +08:00
e3bfd031d8 feat: unified CLI (python -m ai_mouse {train,eval,balabit-adapter})
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-05-10 13:39:20 +08:00
9c4c7f6f0c feat(eval): CLI for generating evaluation reports
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-05-10 13:39:16 +08:00
05be116cde feat(eval): Markdown report builder with matplotlib plots
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-05-10 13:39:13 +08:00
dc38f031b8 feat(eval): kinematics metrics, FFT spectrum, KL divergence
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-05-10 13:33:19 +08:00
3fb4d3a8c0 feat(generator): add 5-point gaussian smoothing on lateral 2026-05-10 13:27:15 +08:00
50dbf40709 refactor(generator): remove deterministic speed_profile and hard log_dt clip
These post-processing hacks were added to compensate for small-data
training. With Balabit pretraining they suppress the multimodal
timing distribution and cause the template-y Δt curves seen in the
verify UI.
2026-05-10 13:22:57 +08:00
b01f6512c0 feat(server): auto-resume from pretrained checkpoint when available 2026-05-10 13:17:24 +08:00
1d9dff4f2a fix(trainer): rename _LOG_1_1 → _LOG_INV_0_9, add tests for aug_ids 2-5 2026-05-10 13:11:27 +08:00
e992823aef feat(trainer): add resume_from for two-stage training 2026-05-10 13:08:37 +08:00
bba8fc567b feat(trainer): replace eager _augment with streaming TrajectoryDataset
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-05-10 12:51:52 +08:00
0b1d18d15c feat(adapter): implement process_session and CLI 2026-05-10 12:42:43 +08:00
cc855e0ae2 feat(adapter): implement segment quality filters 2026-05-10 12:38:17 +08:00
82f803403c feat(adapter): implement click-anchored segmentation 2026-05-10 12:35:43 +08:00
1559f7ab11 feat(adapter): implement Balabit CSV parser 2026-05-10 12:32:20 +08:00
97a835769e feat(adapter): scaffold balabit data adapter package 2026-05-10 12:29:11 +08:00
4d414fd180 chore: initialize git repo, add matplotlib dep, extend config
- Add .gitignore for Python/data/models
- Add matplotlib>=3.8.0 for eval plots
- Add PretrainConfig, FinetuneConfig, BalabitAdapterConfig, EvalConfig dataclasses
2026-05-10 12:24:07 +08:00