universal-dependencies/universal_dependencies
Updated • 4.33k • 36
How to use KoichiYasuoka/roberta-base-korean-upos with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="KoichiYasuoka/roberta-base-korean-upos") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("KoichiYasuoka/roberta-base-korean-upos")
model = AutoModelForTokenClassification.from_pretrained("KoichiYasuoka/roberta-base-korean-upos")This is a RoBERTa model pre-trained on Korean texts for POS-tagging and dependency-parsing, derived from roberta-base-korean-hanja. Every word (어절) is tagged by UPOS(Universal Part-Of-Speech).
from transformers import AutoTokenizer,AutoModelForTokenClassification,TokenClassificationPipeline
tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/roberta-base-korean-upos")
model=AutoModelForTokenClassification.from_pretrained("KoichiYasuoka/roberta-base-korean-upos")
pipeline=TokenClassificationPipeline(tokenizer=tokenizer,model=model,aggregation_strategy="simple")
nlp=lambda x:[(x[t["start"]:t["end"]],t["entity_group"]) for t in pipeline(x)]
print(nlp("홍시 맛이 나서 홍시라 생각한다."))
or
import esupar
nlp=esupar.load("KoichiYasuoka/roberta-base-korean-upos")
print(nlp("홍시 맛이 나서 홍시라 생각한다."))
esupar: Tokenizer POS-tagger and Dependency-parser with BERT/RoBERTa/DeBERTa models
Base model
klue/roberta-base