---
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 
    )