KoichiYasuoka's picture
dependency-parsing
c4d19ac
|
raw
history blame
1.58 kB
metadata
language:
  - ja
tags:
  - japanese
  - token-classification
  - pos
  - dependency-parsing
datasets:
  - universal_dependencies
license: cc-by-sa-4.0
pipeline_tag: token-classification
widget:
  - text: 国境の長いトンネルを抜けると雪国であった。

roberta-small-japanese-char-luw-upos

Model Description

This is a RoBERTa model pre-trained on Japanese 青空文庫 texts for POS-tagging and dependency-parsing, derived from roberta-small-japanese-aozora-char. Every long-unit-word is tagged by UPOS (Universal Part-Of-Speech).

How to Use

from transformers import AutoTokenizer,AutoModelForTokenClassification,TokenClassificationPipeline
tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/roberta-small-japanese-char-luw-upos")
model=AutoModelForTokenClassification.from_pretrained("KoichiYasuoka/roberta-small-japanese-char-luw-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-small-japanese-char-luw-upos")
print(nlp("国境の長いトンネルを抜けると雪国であった。"))

See Also

esupar: Tokenizer POS-tagger and Dependency-parser with BERT/RoBERTa models