KoGrammar
Collection
3 items • Updated
How to use theSOL1/kogrammar-base with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("theSOL1/kogrammar-base")
model = AutoModelForSeq2SeqLM.from_pretrained("theSOL1/kogrammar-base")Dataset: 국립국어원 맞춤법 교정 말뭉치
Backbone Model: kobart-base-v2
GitHub Repo: SOL1archive/KoGrammar
전체 데이터셋 중 약 45%를 학습데이터로 활용하여 학습함.
| BLEU-2 | ROUGE-2 F1 |
|---|---|
| 77.8 % | 55.0 % |