분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
[Private 10th] roberta + preprocessing
공동작성자
1. Preprocessing
- translator를 사용하여 test_data['text']에 있는 각 텍스트에 대해 번역을 적용하였습니다.
2. roberta 모델 사용
- PyTorch와 HuggingFace의 Transformers 라이브러리를 사용하여 RoBERTa 모델을 감정 분석 작업에 맞게 미세 조정하는 작업을 수행하였습니다.
- 모델과 토크나이저: RoBERTa base model (roberta-base)
- 텍스트 처리: 최대 텍스트 길이: 256 토큰 (max_length=256), 입력 텍스트는 truncation과 padding 처리가 되어 있습니다.
- 데이터셋 분할: 훈련 데이터: 90% (train_size = int(0.9 * len(dataset))), 검증 데이터: 10% (나머지)
- Dataloader 설정: 훈련 데이터 배치 크기: 32 (batch_size=32), 검증 데이터 배치 크기: 32 (batch_size=32)
- 옵티마이저: Optimizer - AdamW, Learning rate (lr) - 5e-5
- 훈련 에폭: 3
- GPU 설정: CUDA가 가능한 경우 GPU를 사용하며, 그 중 'Tesla T4'인지 확인하고 출력하고 CUDA가 불가능한 경우 CPU를 사용하였습니다.
DACON Co.,Ltd | CEO Kookjin Kim | 699-81-01021
Mail-order-sales Registration Number: 2021-서울영등포-1704
Business Providing Employment Information Number: J1204020250004
#901, Eunhaeng-ro 3, Yeongdeungpo-gu, Seoul 07237
E-mail dacon@dacon.io |
Tel. 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved