| { | |
| "model_name": "distilbert-base-uncased", | |
| "num_labels": 14, | |
| "emotion_columns": [ | |
| "amusement", | |
| "anger", | |
| "annoyance", | |
| "caring", | |
| "confusion", | |
| "disappointment", | |
| "disgust", | |
| "embarrassment", | |
| "excitement", | |
| "fear", | |
| "gratitude", | |
| "joy", | |
| "love", | |
| "sadness" | |
| ], | |
| "max_length": 128, | |
| "training_args": { | |
| "epochs": 3, | |
| "batch_size": 16, | |
| "learning_rate": 2e-05 | |
| }, | |
| "performance": { | |
| "kaggle_score": 0.4214, | |
| "validation_f1_macro": 0.4275, | |
| "validation_f1_micro": 0.4226, | |
| "hamming_loss": 0.1816, | |
| "jaccard_score": 0.2679 | |
| } | |
| } |