Instructions to use yeongjoon/Kconvo-roberta with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use yeongjoon/Kconvo-roberta with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="yeongjoon/Kconvo-roberta")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("yeongjoon/Kconvo-roberta") model = AutoModelForMaskedLM.from_pretrained("yeongjoon/Kconvo-roberta", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Kconvo-roberta: Korean conversation RoBERTa (github)
- There are many PLMs (Pretrained Language Models) for Korean, but most of them are trained with written language.
- Here, we introduce a retrained PLM for prediction of Korean conversation data where we use verbal data for training.
Usage
# Kconvo-roberta
from transformers import RobertaTokenizerFast, RobertaModel
tokenizer_roberta = RobertaTokenizerFast.from_pretrained("yeongjoon/Kconvo-roberta")
model_roberta = RobertaModel.from_pretrained("yeongjoon/Kconvo-roberta")
Domain Robust Retraining of Pretrained Language Model
- Kconvo-roberta uses klue/roberta-base as the base model and retrained additionaly with the conversation dataset.
- The retrained dataset was collected through the National Institute of the Korean Language and AI-Hub, and the collected dataset is as follows.
- National Institute of the Korean Language
* μ¨λΌμΈ λν λ§λμΉ 2021
* μΌμ λν λ§λμΉ 2020
* κ΅¬μ΄ λ§λμΉ
* λ©μ μ λ§λμΉ
- AI-Hub
* μ¨λΌμΈ ꡬμ΄μ²΄ λ§λμΉ λ°μ΄ν°
* μλ΄ μμ±
* νκ΅μ΄ μμ±
* μμ λν μμ±(μΌλ°λ¨μ¬)
* μΌμμν λ° κ΅¬μ΄μ²΄ ν-μ λ²μ λ³λ ¬ λ§λμΉ λ°μ΄ν°
* νκ΅μΈ λνμμ±
* κ°μ± λν λ§λμΉ
* μ£Όμ λ³ ν
μ€νΈ μΌμ λν λ°μ΄ν°
* μ©λλ³ λͺ©μ λν λ°μ΄ν°
* νκ΅μ΄ SNS
- Downloads last month
- 7