diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/base_model_trpgner/__init__.py (renamed from src/basemodel/__init__.py) | 14 | ||||
| -rw-r--r-- | src/base_model_trpgner/download_model.py (renamed from src/basemodel/download_model.py) | 0 | ||||
| -rw-r--r-- | src/base_model_trpgner/inference/__init__.py (renamed from src/basemodel/inference/__init__.py) | 0 | ||||
| -rw-r--r-- | src/base_model_trpgner/training/__init__.py (renamed from src/basemodel/training/__init__.py) | 4 | ||||
| -rw-r--r-- | src/base_model_trpgner/utils/__init__.py (renamed from src/basemodel/utils/__init__.py) | 0 |
5 files changed, 9 insertions, 9 deletions
diff --git a/src/basemodel/__init__.py b/src/base_model_trpgner/__init__.py index 7287df4..9796c83 100644 --- a/src/basemodel/__init__.py +++ b/src/base_model_trpgner/__init__.py @@ -1,28 +1,28 @@ """ -base-model - HydroRoll TRPG NER 模型 SDK +base-model-trpgner - HydroRoll TRPG NER 模型 SDK 这是一个用于 TRPG(桌上角色扮演游戏)日志命名实体识别的 Python SDK。 基本用法: - >>> from basemodel import TRPGParser + >>> 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[train] - >>> from basemodel.training import train_ner_model + >>> 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 basemodel.inference import TRPGParser, parse_line, parse_lines +from base_model_trpgner.inference import TRPGParser, parse_line, parse_lines try: from importlib.metadata import version - __version__ = version("base-model") + __version__ = version("base_model_trpgner") except Exception: - __version__ = "0.1.0.dev" + __version__ = "0.1.1.dev" __all__ = [ "__version__", diff --git a/src/basemodel/download_model.py b/src/base_model_trpgner/download_model.py index 2d65099..2d65099 100644 --- a/src/basemodel/download_model.py +++ b/src/base_model_trpgner/download_model.py diff --git a/src/basemodel/inference/__init__.py b/src/base_model_trpgner/inference/__init__.py index 93a185f..93a185f 100644 --- a/src/basemodel/inference/__init__.py +++ b/src/base_model_trpgner/inference/__init__.py diff --git a/src/basemodel/training/__init__.py b/src/base_model_trpgner/training/__init__.py index 5671c42..ccf3c03 100644 --- a/src/basemodel/training/__init__.py +++ b/src/base_model_trpgner/training/__init__.py @@ -36,7 +36,7 @@ def train_ner_model( resume_from_checkpoint: 恢复检查点路径 Examples: - >>> from basemodel.training import train_ner_model + >>> from basemodeltrpgner.training import train_ner_model >>> train_ner_model( ... conll_data="./data", ... output_dir="./my_model", @@ -59,7 +59,7 @@ def train_ner_model( ) from e # 导入数据处理函数 - from basemodel.utils.conll import load_conll_dataset, tokenize_and_align_labels + from base_model_trpgner.utils.conll import load_conll_dataset, tokenize_and_align_labels print(f"🚀 Starting training...") diff --git a/src/basemodel/utils/__init__.py b/src/base_model_trpgner/utils/__init__.py index 12a3ef4..12a3ef4 100644 --- a/src/basemodel/utils/__init__.py +++ b/src/base_model_trpgner/utils/__init__.py |
