Downloads · 30 days
9
0% of all-time downloads
kor-naver-ner-name
joon09/kor-naver-ner-name
kor-naver-ner-name is a token classification model from joon09. Use it when you need labels on individual words, such as names. It is set up for transformers.
All-time downloads
511K
Public
Repo size
1.4 GB
Likes
4
Public
Repo makeup
Click a slice to open those files.
.bin471 MB · 100%
At a glance
- Task
- Token Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Aug 7, 2022
- Updated
- Apr 4, 2023
- SHA
- 2d8f6b4b
Try a prompt
Model card
From the Hugging Face model README
한국인 이름 인식 모델
kor-bert fine-tuning 모델
자주 안쓰는 한글이름 기준으로 생성기를 만들어서, 16만개의 한글 이름을 생성 후 학습한 모델입니다.
ex) 안녕하세요. 임준영입니다. -> 안녕하세요. ***입니다.
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}]
- Task
- Token Classification
- Library
- transformers
- Type
- bert
- Created
- Aug 7, 2022
- Updated
- Apr 4, 2023