2. BERT Multi Classification 모델구현
1) 학습 및 평가 데이터 세트
(1) 데이터 세트 준비
(2) 데이터 세트 클래스 정의
2) Pre-trained 모델 다운로드 및 Fine-tunning 설정
(1) Pre-trained 모델 다운로드
- num_labels은 멀티 classification에 반드시 입력해야 함.
(2) Training을 위한 하이퍼 파라미터 설정
반응형
'🥾 프로젝트 > (STEP)파이썬을 활용한 인공지능 자연어 처리' 카테고리의 다른 글
| 8-1회차) BERT 모델을 활용한 Text Classification 모델 구현 (0) | 2023.05.15 |
|---|---|
| 7-2회차) 자연어 처리를 위한 BERT 모델 (1) | 2023.05.14 |
| 7-1회차) 자연어 처리를 위한 BERT 모델 (0) | 2023.05.13 |
| 6-2회차) 자연어 처리를 위한 Transformer 적용 (0) | 2023.05.11 |
| 6-1회차) 자연어 처리를 위한 Transformer 적용 (0) | 2023.05.10 |
