diff --git a/config.json b/config.json index ce5d3a6..1575d82 100644 --- a/config.json +++ b/config.json @@ -22,13 +22,13 @@ "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { - "LABEL_0": 0, - "LABEL_1": 1, - "LABEL_2": 2, - "LABEL_3": 3, - "LABEL_4": 4, - "LABEL_5": 5, - "LABEL_6": 6 + "anger": 0, + "disgust": 1, + "fear": 2, + "joy": 3, + "neutral": 4, + "sadness": 5, + "surprise": 6 }, "layer_norm_eps": 1e-05, "max_position_embeddings": 514,