From 8298d8a51e716e2f98b988551a604423c6a4df1a Mon Sep 17 00:00:00 2001 From: j-hartmann Date: Tue, 15 Jun 2021 10:16:23 +0000 Subject: [PATCH] updated labels --- config.json | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) 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,