BERTc-165M-MT / example_decode.py
tf-bao's picture
Upload BERTc-165M-MT release
866611a verified
Raw
History Blame Contribute Delete
247 Bytes
from mt_model import BERTcForMT, PieceCharTokenizer
tok = PieceCharTokenizer(".")
model = BERTcForMT.from_pretrained(".")
result = model.predict("中华人民共和国万岁", tok)
print(result["words"])
print(result["pos"])
print(result["ner"])