diff --git a/config.json b/config.json index 10ea5f2..ce5d3a6 100644 --- a/config.json +++ b/config.json @@ -11,13 +11,13 @@ "hidden_dropout_prob": 0.1, "hidden_size": 768, "id2label": { - "0": "LABEL_0", - "1": "LABEL_1", - "2": "LABEL_2", - "3": "LABEL_3", - "4": "LABEL_4", - "5": "LABEL_5", - "6": "LABEL_6" + "0": "anger", + "1": "disgust", + "2": "fear", + "3": "joy", + "4": "neutral", + "5": "sadness", + "6": "surprise" }, "initializer_range": 0.02, "intermediate_size": 3072,