metadata

language:
  - th
tags:
  - thai
  - token-classification
  - pos
  - wikipedia
license: apache-2.0
pipeline_tag: token-classification
widget:
  - text: หลายหัวดีกว่าหัวเดียว

bert-base-thai-upos

Model Description

This is a BERT model pre-trained on Thai Wikipedia texts for POS-tagging, derived from bert-base-th-cased. Every word is tagged by UPOS (Universal Part-Of-Speech).

How to Use

import torch
from transformers import AutoTokenizer,AutoModelForTokenClassification
tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/bert-base-thai-upos")
model=AutoModelForTokenClassification.from_pretrained("KoichiYasuoka/bert-base-thai-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)))