DistilBERT IMDb Sentiment Analysis

학습 상세

• 베이스 모델: distilbert-base-uncased
• 데이터셋: IMDb (train 2,000 / test 500 샘플)
• 태스크: 이진 감정분류 (긍정/부정)
• Epoch: 2
• Learning Rate: 2e-5
• Batch Size: 16
• Max Length: 256

사용법

한계

• 학습 데이터를 2,000개로 축소하여 실습 목적으로 학습함
• 전체 데이터(25K)로 학습 시 성능 향상 기대
• 영어 리뷰 전용