| { | |
| "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 | |
| } | |
| } | |