월간 데이콘 한국어 문장 관계 분류 경진대회

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

  • moneyIcon 상금 : 100만원 + α
  • 1,342명 마감

 

허깅페이스 학습 파이프라인(klue/roberta-base, LB 0.805)

2022.02.06 20:15 5,859 조회 language

klue/bert-base를 기반으로 분류(contradiction, neutral, entailment)를 예측하는 코드입니다. 코드를 huggingface라이브러리 기반으로 작성하여, 비교적 간단하게 구현하였습니다. 간단한 분류 모델을 학습하는 데에 도움이 되었으면 좋겠습니다.

코드
로그인이 필요합니다
0 / 1000
palette
2022.02.07 10:38

감사합니다!! XD

arethusa
2022.02.09 09:22

감사합니다:)

SeoroMin
2022.02.08 13:57

감사합니다! lr같은 하이퍼파라미터는 직접 조정 못하는 건가요??

arethusa
2022.02.09 09:22

TrainingArguments를 기반으로 조정 가능합니다.

애뚱
2022.02.09 14:15

좋은 글 감사합니다.
혹시 해당 코드는 로컬에서 돌리신걸까요? 
구글 colab에서 돌리니 "CUDA error: device-side assert triggered" 에러가 발생해서요 ㅠ

arethusa
2022.02.09 23:04

코랩에서 돌렸고, 해당 에러는 입출력 처리가 잘못되어서 발생하는 에러인데, 여기서 제공하는 것 그대로 사용하시면 에러가 없을 듯 합니다...ㅠ