ETRI 한국어 BERT모델 활용한 text classification

ETRI 에서 제공하는 한국어 BERT 모델인 KorBERT를 활용하여 한국어 text classification 모델을 Pytorch 로 학습할 수 있도록 하였습니다.

ETRI 의 한국어 BERT 모델은 해당 링크에서 서약서를 작성한 후 키 발급을 통해 다운 받을 수 있습니다.

Dataset

Naver sentiment movie corpus v1.0

Requirement

• Python 3.7
• Pytorch == 1.5.0
• urllib3==1.24.3
• boto3==1.9.66
• botocore==1.12.67
• requests==2.24.0
• tqdm==4.50.2

Models

Results

Model	MLP	CNN	LSTM	Bi-LSTM
Accuracy