| { | |
| "architectures": ["ElectraForMultiLabelClassification"], | |
| "attention_probs_dropout_prob": 0.1, | |
| "embedding_size": 128, | |
| "finetuning_task": "goemotions", | |
| "hidden_act": "gelu", | |
| "hidden_dropout_prob": 0.1, | |
| "hidden_size": 256, | |
| "id2label": { | |
| "0": "admiration", | |
| "1": "amusement", | |
| "10": "disapproval", | |
| "11": "disgust", | |
| "12": "embarrassment", | |
| "13": "excitement", | |
| "14": "fear", | |
| "15": "gratitude", | |
| "16": "grief", | |
| "17": "joy", | |
| "18": "love", | |
| "19": "nervousness", | |
| "2": "anger", | |
| "20": "optimism", | |
| "21": "pride", | |
| "22": "realization", | |
| "23": "relief", | |
| "24": "remorse", | |
| "25": "sadness", | |
| "26": "surprise", | |
| "27": "neutral", | |
| "3": "annoyance", | |
| "4": "approval", | |
| "5": "caring", | |
| "6": "confusion", | |
| "7": "curiosity", | |
| "8": "desire", | |
| "9": "disappointment" | |
| }, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 1024, | |
| "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, | |
| "neutral": 27, | |
| "optimism": 20, | |
| "pride": 21, | |
| "realization": 22, | |
| "relief": 23, | |
| "remorse": 24, | |
| "sadness": 25, | |
| "surprise": 26 | |
| }, | |
| "layer_norm_eps": 1e-12, | |
| "max_position_embeddings": 512, | |
| "model_type": "electra", | |
| "num_attention_heads": 4, | |
| "num_hidden_layers": 12, | |
| "pad_token_id": 0, | |
| "type_vocab_size": 2, | |
| "vocab_size": 32200 | |
| } | |