bert-base-japanese-upos

模型描述

这是一个在日语维基百科文本上进行POS标注和依存解析的BERT模型，衍生自 bert-base-japanese-char-extended 。每个短单位词都被标记为 UPOS (Universal Part-Of-Speech)。

如何使用

import torch
from transformers import AutoTokenizer,AutoModelForTokenClassification
tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/bert-base-japanese-upos")
model=AutoModelForTokenClassification.from_pretrained("KoichiYasuoka/bert-base-japanese-upos")
s="国境の長いトンネルを抜けると雪国であった。"
p=[model.config.id2label[q] for q in torch.argmax(model(tokenizer.encode(s,return_tensors="pt"))["logits"],dim=2)[0].tolist()[1:-1]]
print(list(zip(s,p)))

或者

import esupar
nlp=esupar.load("KoichiYasuoka/bert-base-japanese-upos")
print(nlp("国境の長いトンネルを抜けると雪国であった。"))

参见

esupar ：使用BERT/RoBERTa/DeBERTa模型的分词器POS标注器和依存解析器