월간 데이콘 뉴스 토픽 분류 AI 경진대회

알고리즘 | NLP | 분류 | 자연어 | Accuracy

  • moneyIcon Prize : 500,000 D-point
  • 1,795명 마감

 

[Private 2nd] Huggingface를 사용한 베이스라인

2021.08.12 21:26 13,809 Views language

7 models Ensemble한 Simple baseline 입니다

# Details
Training : 빠른 학습속도를 위한 AMP 사용
Epoch : 10
Data : 각 모델별로 5 Fold 학습
Model :  'monologg/kobert',  'klue/roberta-base',  'klue/roberta-small',  'klue/roberta-large',  'xlm-roberta-large',  'bert-base-multilingual-uncased'
Ensemble : logit ensemble

Code
로그인이 필요합니다
0 / 1000
DACON.tuna
2021.08.22 23:37

각 모델이 엄청 무거울 것 같네요!
축하드립니다. 그리고 코드 공유해 주셔서 감사합니다!

samkim
2021.09.20 13:37

축하드립니다! 그리고 좋은 코드 공유 감사드립니다
혹시 사용된 라이브러리들의 버전을 알 수 있을까요? 제가 현재 사용하는 torch와 조금 다른 것 같아서요

kobi2000
2022.10.07 21:11

batch 사이즈를 1로 하셨나요?

용맹한호랑이
2023.07.30 14:12

KeyError: "None of [Index(['title', 'topic_idx'], dtype='object')] are in the [columns]"
가 계속 출력되는데 어떻게 해결하면 될까요?