From 5dd166366b8a2f4699c1841ebd7fceabcd9868a4 Mon Sep 17 00:00:00 2001 From: HsiangNianian Date: Tue, 30 Dec 2025 20:16:05 +0800 Subject: refactor: Refactor TRPG NER model SDK: restructure codebase into base_model_trpgner package, implement training and inference modules, and add model download functionality. Remove legacy training and utils modules. Enhance documentation and examples for better usability. --- src/base_model_trpgner/__init__.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 src/base_model_trpgner/__init__.py (limited to 'src/base_model_trpgner/__init__.py') diff --git a/src/base_model_trpgner/__init__.py b/src/base_model_trpgner/__init__.py new file mode 100644 index 0000000..9796c83 --- /dev/null +++ b/src/base_model_trpgner/__init__.py @@ -0,0 +1,36 @@ +""" +base-model-trpgner - HydroRoll TRPG NER 模型 SDK + +这是一个用于 TRPG（桌上角色扮演游戏）日志命名实体识别的 Python SDK。 + +基本用法: + >>> from base_model_trpgner import TRPGParser + >>> parser = TRPGParser() + >>> result = parser.parse("风雨 2024-06-08 21:44:59 剧烈的疼痛...") + >>> print(result) + {'metadata': {'speaker': '风雨', 'timestamp': '2024-06-08 21:44:59'}, 'content': [...]} + +训练功能（需要额外安装）: + >>> pip install base-model-trpgner[train] + >>> from base_model_trpgner.training import train_ner_model + >>> train_ner_model(conll_data="./data", output_dir="./model") +""" + +from base_model_trpgner.inference import TRPGParser, parse_line, parse_lines + +try: + from importlib.metadata import version + __version__ = version("base_model_trpgner") +except Exception: + __version__ = "0.1.1.dev" + +__all__ = [ + "__version__", + "TRPGParser", + "parse_line", + "parse_lines", +] + + +def get_version(): + return __version__ -- cgit v1.2.3-70-g09d2