ckiplab / ckip-transformers

CKIP Transformers

Home Page:https://ckip-transformers.readthedocs.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

NER 怎么使用输出ner对应标签

lonngxiang opened this issue · comments

from transformers import (
BertTokenizerFast,
AutoModel,
)

tokenizer = BertTokenizerFast.from_pretrained('bert-base-chinese')
model = AutoModel.from_pretrained('ckiplab/bert-base-chinese-ner')

。。。
这后面应该怎么写呢,才能对应输出这种
image

關於標記的轉換,可使用 CKIP Transformers 的 NLP Tools 進行轉換。

至於詳細的 ID 對應,可參見 huggingface 的 model config 檔案 (如 https://huggingface.co/ckiplab/bert-base-chinese-ner/blob/main/config.json)。