"""BERTc-MT:分词 + 词性 + 命名实体。"""
from mt_model import BERTcForMT

model = BERTcForMT.from_pretrained(".")
for r in model.predict([
    "中国科学院计算技术研究所在北京",
    "李雷和韩梅梅去上海参加了会议",
]):
    print(r["text"])
    print("  分词:", " / ".join(r["words"]))
    print("  词性:", " ".join(f"{w}/{p}" for w, p in zip(r["words"], r["pos"])))
    print("  实体:", r["ner"])