NER_Game / example.py
li-ovo's picture
Update NER_Game to R27 merged multi-game BF16 release
5c86bf0 verified
Raw History Blame Contribute Delete
708 Bytes
"""R27 多游戏抽取示例:下载仓库后执行 python example.py。"""
import json
import sys
from pathlib import Path
root = Path(__file__).resolve().parent
sys.path.insert(0, str(root / "sdk"))
from extraction_sdk import SemanticUnitExtractor
extractor = SemanticUnitExtractor(str(root / "model"), dtype="bf16")
records = [
{"id": "demo-001", "text": "孙悟空打野", "game_name": "王者荣耀"},
{"id": "demo-002", "text": "原神抽卡,原神更新后卡顿", "game_name": "原神"},
{"id": "demo-003", "text": "和平精英组队排位", "game_name": "和平精英"},
]
for result in extractor.extract_batch(records):
print(json.dumps(result, ensure_ascii=False, indent=2))