Mizuiro-sakura's picture
Update README.md
54f1751
|
raw
history blame
2.96 kB
---
license: mit
language: ja
library_name: transformers
tags:
- pytorch
- deberta
- deberta-v2
- named entity recognition
- named-entity-recognition
- ner
datasets:
- wikipedia
- cc100
- oscar
metrics:
- accuracy
---
# このモデルはdeberta-v2-base-japaneseをファインチューニングして固有表現抽出(NER)に用いれるようにしたものです。
このモデルはdeberta-v2-base-japaneseを Wikipediaを用いた日本語の固有表現抽出データセット(ストックマーク社、https://github.com/stockmarkteam/ner-wikipedia-dataset )を用いてファインチューニングしたものです。
# This model is fine-tuned model for Named Entity Recognition (NER) which is based on deberta-v2-base-japanese
This model is fine-tuned by using Wikipedia dataset.
You could use this model for NER tasks.
# How to use 使い方
transformersおよびpytorch、sentencepiece、Juman++をインストールしてください。
以下のコードを実行することで、固有表現抽出タスクを解かせることができます。 please execute this code.
```python
from transformers import AutoTokenizer,pipeline, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained('Mizuiro-sakura/deberta-v2-base-japanese-finetuned-ner')
model=AutoModelForTokenClassification.from_pretrained('Mizuiro-sakura/deberta-v2-base-japanese-finetuned-ner') # 学習済みモデルの読み込み
text=('昨日は東京で買い物をした')
ner=pipeline('ner', model=model, tokenizer=tokenizer)
result=ner(text)
print(result)
```
# モデルの精度 accuracy of model
precision recall f1-score support
その他の組織名 0.73 0.75 0.74 238
イベント名 0.81 0.81 0.81 215
人名 0.84 0.87 0.85 547
地名 0.83 0.83 0.83 446
政治的組織名 0.82 0.85 0.83 263
施設名 0.74 0.86 0.80 241
法人名 0.81 0.82 0.82 487
製品名 0.68 0.73 0.71 252
micro avg 0.79 0.82 0.81 2689
macro avg 0.78 0.81 0.80 2689
weighted avg 0.79 0.82 0.81 2689
# deberta-v2-base-japaneseとは?
日本語Wikipedeia(3.2GB)および、cc100(85GB)、oscar(54GB)を用いて訓練されたモデルです。
京都大学黒橋研究室が公表されました。
# Model description
This is a Japanese DeBERTa V2 base model pre-trained on Japanese Wikipedia, the Japanese portion of CC-100, and the Japanese portion of OSCAR.
# Acknowledgments 謝辞
モデルを公開してくださった京都大学黒橋研究室には感謝いたします。
I would like to thank Kurohashi Lab at Kyoto University.