base-model/tests, branch v0.1.1BaseModel for HydroRoll https://git.hydroroll.team/base-model/atom?h=v0.1.12025-12-30T11:14:39Zfeat: Implement TRPG NER training and inference script with robust model path detection and enhanced timestamp/speaker handling2025-12-30T11:14:39ZHsiangNianiani@jyunko.cn2025-12-30T11:14:39Zurn:sha1:7ac684f1f82023c6284cd7d7efde11b8dc98c149
- Added main training and inference logic in main.py, including CoNLL parsing, tokenization, and model training.
- Introduced TRPGParser class for inference with entity aggregation and special handling for timestamps and speakers.
- Developed utility functions for converting word-level CoNLL to char-level and saving datasets in various formats.
- Added ONNX export functionality for the trained model.
- Created a comprehensive requirements.txt and updated pyproject.toml with necessary dependencies.
- Implemented tests for ONNX inference to validate model outputs.