Emanuel lewtun HF staff commited on
Commit
468fc00
1 Parent(s): ef95353

Align label mapping with emotion dataset (#1)

Browse files

- Align label mapping with emotion dataset (502cba811da81b2df584ce9d150f53ba7c3786c2)


Co-authored-by: Lewis Tunstall <[email protected]>

Files changed (1) hide show
  1. config.json +13 -13
config.json CHANGED
@@ -9,22 +9,22 @@
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1",
14
- "2": "LABEL_2",
15
- "3": "LABEL_3",
16
- "4": "LABEL_4",
17
- "5": "LABEL_5"
18
  },
19
  "initializer_range": 0.02,
20
  "intermediate_size": 3072,
21
  "label2id": {
22
- "LABEL_0": 0,
23
- "LABEL_1": 1,
24
- "LABEL_2": 2,
25
- "LABEL_3": 3,
26
- "LABEL_4": 4,
27
- "LABEL_5": 5
28
  },
29
  "layer_norm_eps": 1e-07,
30
  "max_position_embeddings": 512,
@@ -49,4 +49,4 @@
49
  "transformers_version": "4.13.0.dev0",
50
  "type_vocab_size": 0,
51
  "vocab_size": 128100
52
- }
 
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "sadness",
13
+ "1": "joy",
14
+ "2": "love",
15
+ "3": "anger",
16
+ "4": "fear",
17
+ "5": "surprise"
18
  },
19
  "initializer_range": 0.02,
20
  "intermediate_size": 3072,
21
  "label2id": {
22
+ "sadness": 0,
23
+ "joy": 1,
24
+ "love": 2,
25
+ "anger": 3,
26
+ "fear": 4,
27
+ "surprise": 5
28
  },
29
  "layer_norm_eps": 1e-07,
30
  "max_position_embeddings": 512,
 
49
  "transformers_version": "4.13.0.dev0",
50
  "type_vocab_size": 0,
51
  "vocab_size": 128100
52
+ }