KoichiYasuoka
/

roberta-base-thai-syllable-upos

Token Classification

dependency-parsing

Model card Files Files and versions

KoichiYasuoka commited on Dec 7, 2021

Commit

3dd7ca3

·

1 Parent(s): dcea747

usage improved

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/roberta-base-thai-syllabl
 model=AutoModelForTokenClassification.from_pretrained("KoichiYasuoka/roberta-base-thai-syllable-upos")
 s="หลายหัวดีกว่าหัวเดียว"
 t=tokenizer.tokenize(s)
-p=[model.config.id2label[q] for q in torch.argmax(model(tokenizer.encode(s,return_tensors="pt"))[0],dim=2)[0].tolist()[1:-1]]
 print(list(zip(t,p)))
 ```

 model=AutoModelForTokenClassification.from_pretrained("KoichiYasuoka/roberta-base-thai-syllable-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)))
 ```