yellowdog/ai_mouse
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
69 Commits 1 Branch 1 Tag
adc46a445f047a18e1df3e05ed2776d478a88198
Commit Graph

3 Commits

Author SHA1 Message Date
dog
3e7a194356 docs: spec for mouse post-processing rework + Heun sampling 
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
 2026-07-09 17:11:07 +08:00
Huang Qi
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
Huang Qi
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