模型:
KoichiYasuoka/roberta-base-thai-spm-upos
这是一个在泰国维基百科文本上进行 POS 标注和依存分析的 RoBERTa 模型,派生自 roberta-base-thai-spm 。每个词都经过 UPOS (通用词性标记)进行标记。
import torch from transformers import AutoTokenizer,AutoModelForTokenClassification tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/roberta-base-thai-spm-upos") model=AutoModelForTokenClassification.from_pretrained("KoichiYasuoka/roberta-base-thai-spm-upos") s="หลายหัวดีกว่าหัวเดียว" t=tokenizer.tokenize(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(t,p)))
或者
import esupar nlp=esupar.load("KoichiYasuoka/roberta-base-thai-spm-upos") print(nlp("หลายหัวดีกว่าหัวเดียว"))
esupar : 使用 BERT/RoBERTa/DeBERTa 模型的分词器、词性标注器和依存分析器