Views
No views yet
1import torch
2from transformers import AutoTokenizer,AutoModelForTokenClassification
3tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/deberta-base-japanese-upos")
4model=AutoModelForTokenClassification.from_pretrained("KoichiYasuoka/deberta-base-japanese-upos")
5s="国境の長いトンネルを抜けると雪国であった。"
6t=tokenizer.tokenize(s)
7p=[model.config.id2label[q] for q in torch.argmax(model(tokenizer.encode(s,return_tensors="pt"))["logits"],dim=2)[0].tolist()[1:-1]]
8print(list(zip(t,p)))1import esupar
2nlp=esupar.load("KoichiYasuoka/deberta-base-japanese-upos")
3print(nlp("国境の長いトンネルを抜けると雪国であった。"))