File size: 954 Bytes
d2533a4 69acd94 43caf93 69acd94 52f2302 cf75e25 092ffda 8f8e459 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
---
license: apache-2.0
language:
- en
base_model:
- answerdotai/ModernBERT-base
pipeline_tag: text-classification
tags:
- emotion
- psychology
- finetune
- paulekman
---
label_map = {0: 'sadness', 1: 'joy', 2: 'love', 3: 'anger', 4: 'fear', 5: 'surprise'}
final_training_args = TrainingArguments(
output_dir=outputs_dir,
per_device_train_batch_size=32,
num_train_epochs=3,
weight_decay=0.01,
learning_rate=3e-6,
warmup_steps=700,
lr_scheduler_type="cosine",
eval_strategy="steps",
eval_steps=100,
save_steps=100,
save_strategy="steps",
logging_dir=logs_dir,
logging_steps=100,
load_best_model_at_end=True,
metric_for_best_model="eval_loss",
# report_to="none",
fp16=True,
disable_tqdm=False,
max_grad_norm=10.2
)
 |