Mizuiro-sakura's picture
Update README.md
54f1751
metadata
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.

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.