diff --git a/config.json b/config.json index ce07849..10d8532 100644 --- a/config.json +++ b/config.json @@ -10,17 +10,26 @@ "hidden_act": "gelu", "hidden_dropout_prob": 0.1, "hidden_size": 768, - "id2label": [ - "anger", - "anger", - "anger", - "anger", - "anger", - "anger", - "anger" - ], + "id2label": { + "0": "anger", + "1": "disgust", + "2": "fear", + "3": "joy", + "4": "neutral", + "5": "sadness", + "6": "surprise" + }, "initializer_range": 0.02, "intermediate_size": 3072, + "label2id": { + "anger": 0, + "disgust": 1, + "fear": 2, + "joy": 3, + "neutral": 4, + "sadness": 5, + "surprise": 6 + }, "layer_norm_eps": 1e-05, "max_position_embeddings": 514, "model_type": "roberta", @@ -33,4 +42,4 @@ "type_vocab_size": 1, "use_cache": true, "vocab_size": 50265 -} +} \ No newline at end of file