| { | |
| "finetuned_from": "bert-base-uncased", | |
| "architectures": ["BertForSequenceClassification"], | |
| "num_labels": 28, | |
| "problem_type": "multi_label_classification", | |
| "batch_size": 16, | |
| "max_length": 128, | |
| "epochs": 5, | |
| "learning_rate": 2e-05, | |
| "weight_decay": 0.01, | |
| "focal_loss_alpha": 1, | |
| "focal_loss_gamma": 2, | |
| "id2label": { | |
| "0": "admiration", | |
| "1": "amusement", | |
| "2": "anger", | |
| "3": "annoyance", | |
| "4": "approval", | |
| "5": "caring", | |
| "6": "confusion", | |
| "7": "curiosity", | |
| "8": "desire", | |
| "9": "disappointment", | |
| "10": "disapproval", | |
| "11": "disgust", | |
| "12": "embarrassment", | |
| "13": "excitement", | |
| "14": "fear", | |
| "15": "gratitude", | |
| "16": "grief", | |
| "17": "joy", | |
| "18": "love", | |
| "19": "nervousness", | |
| "20": "optimism", | |
| "21": "pride", | |
| "22": "realization", | |
| "23": "relief", | |
| "24": "remorse", | |
| "25": "sadness", | |
| "26": "surprise", | |
| "27": "neutral" | |
| }, | |
| "label2id": { | |
| "admiration": 0, | |
| "amusement": 1, | |
| "anger": 2, | |
| "annoyance": 3, | |
| "approval": 4, | |
| "caring": 5, | |
| "confusion": 6, | |
| "curiosity": 7, | |
| "desire": 8, | |
| "disappointment": 9, | |
| "disapproval": 10, | |
| "disgust": 11, | |
| "embarrassment": 12, | |
| "excitement": 13, | |
| "fear": 14, | |
| "gratitude": 15, | |
| "grief": 16, | |
| "joy": 17, | |
| "love": 18, | |
| "nervousness": 19, | |
| "optimism": 20, | |
| "pride": 21, | |
| "realization": 22, | |
| "relief": 23, | |
| "remorse": 24, | |
| "sadness": 25, | |
| "surprise": 26, | |
| "neutral": 27 | |
| } | |
| } |