본문 바로가기

🥾 프로젝트/(STEP)파이썬을 활용한 인공지능 자연어 처리

8-2회차) BERT 모델을 활용한 Text Classification 모델 구현

2. BERT Multi Classification 모델구현

1) 학습 및 평가 데이터 세트

 (1) 데이터 세트 준비

 (2) 데이터 세트 클래스 정의

2) Pre-trained 모델 다운로드 및 Fine-tunning 설정

 (1) Pre-trained 모델 다운로드

- num_labels은 멀티 classification에 반드시 입력해야 함.

 

 (2) Training을 위한 하이퍼 파라미터 설정