pip을 이용해 설치
pip install "git+https://github.com/2021-hknu-cd-hate-speech-classification/model.git"from hate_speech_classification_model import HateSpeechClassifier__init__ 함수의 hyper_parameter 매개 변수에 dict형태로 넣어주면 사용할 수 있다.
| 이름 | 설명 | 값으로 들어갈 수 있는 것 | 기본값 |
|---|---|---|---|
model |
fine-tuning에 사용될 모델 이름 | 'beomi/KcELECTRA', 'monologg/koelectra-base-v3-discriminator' |
beomi/KcELECTRA |
max_length |
Tokenizer가 사용할 문자열 길이 | int형 값. |
150 |
lr |
학습률 | float형 값. |
5e-6 |
epochs |
epoch 수 | int형 값. |
5 |
optimizer |
사용할 optimizer의 종류 | 'adam', 'adamw', 'sgd' |
adamw |
gamma |
Scheduler가 사용할 gamma값 | float형 값. |
0.5 |
batch_size |
DataLoader가 사용할 batch_size 값 | int형 값 |
32 |