diff --git a/config.json b/config.json index 1575d82..ce07849 100644 --- a/config.json +++ b/config.json @@ -10,26 +10,17 @@ "hidden_act": "gelu", "hidden_dropout_prob": 0.1, "hidden_size": 768, - "id2label": { - "0": "anger", - "1": "disgust", - "2": "fear", - "3": "joy", - "4": "neutral", - "5": "sadness", - "6": "surprise" - }, + "id2label": [ + "anger", + "anger", + "anger", + "anger", + "anger", + "anger", + "anger" + ], "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",