Kor Naver Ner Name

🚀 韓国人の名前認識モデル

このモデルはkor-bertをfine-tuningしたもので、あまり使われない韓国語の名前を基準に生成器を作成し、16万個の韓国語の名前を生成して学習させたモデルです。

例えば、「안녕하세요. 임준영입니다.」という文を入力すると、「안녕하세요. ***입니다.」のように名前部分をマスキングします。

🚀 クイックスタート

このモデルを使用するには、以下の手順で操作します。

💻 使用例

基本的な使用法

from transformers import BertTokenizerFast, BertForTokenClassification
from transformers import pipeline

model_name = 'joon09/kor-naver-ner-name'
tokenizer = BertTokenizerFast.from_pretrained(model_name)
model = BertForTokenClassification.from_pretrained(model_name)
nlp = pipeline("ner", model=model, tokenizer=tokenizer)

ner('안녕하세요. 임준영입니다.',grouped_entities=True,aggregation_strategy='average')

[{'entity_group': 'PER',
  'score': 0.99999785,
  'word': '임',
  'start': 7,
  'end': 8},
 {'entity_group': 'PER',
  'score': 0.82035744,
  'word': '##준영',
  'start': 8,
  'end': 10}]