{
  "model_type": "segue-multitask",
  "base_model": "declare-lab/segue-w2v2-base",
  "tasks": {
    "sentiment": {"num_labels": 3, "labels": ["neutral", "positive", "negative"]},
    "emotion":   {"num_labels": 7, "labels": ["neutral", "surprise", "fear", "sadness", "joy", "disgust", "anger"]}
  },
  "dataset": "MELD",
  "training": {
    "learning_rate": 3e-5,
    "num_epochs": 3,
    "warmup_ratio": 0.3,
    "batch_size": 1,
    "gradient_accumulation_steps": 8,
    "avg_checkpoints": 10,
    "save_steps": 100,
    "seed": 39
  },
  "results": {
    "test_sentiment_weighted_f1": 0.5582,
    "test_sentiment_macro_f1": 0.5185,
    "test_emotion_weighted_f1": 0.4750,
    "test_emotion_macro_f1": 0.2733
  }
}