From 599d32d232cba93389032b255a4f49140e538daa Mon Sep 17 00:00:00 2001 From: j-hartmann Date: Tue, 15 Jun 2021 10:29:09 +0000 Subject: [PATCH] updated labels --- config.json | 27 +++++++++------------------ 1 file changed, 9 insertions(+), 18 deletions(-) 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",