xbgoose commited on
Commit
22646d5
1 Parent(s): a3bd41d

Update config.json

Browse files

Update id2label and label2id mapping

Files changed (1) hide show
  1. config.json +10 -10
config.json CHANGED
@@ -53,20 +53,20 @@
53
  "hidden_dropout_prob": 0.1,
54
  "hidden_size": 1024,
55
  "id2label": {
56
- "0": "LABEL_0",
57
- "1": "LABEL_1",
58
- "2": "LABEL_2",
59
- "3": "LABEL_3",
60
- "4": "LABEL_4"
61
  },
62
  "initializer_range": 0.02,
63
  "intermediate_size": 4096,
64
  "label2id": {
65
- "LABEL_0": 0,
66
- "LABEL_1": 1,
67
- "LABEL_2": 2,
68
- "LABEL_3": 3,
69
- "LABEL_4": 4
70
  },
71
  "layer_norm_eps": 1e-05,
72
  "layerdrop": 0.1,
 
53
  "hidden_dropout_prob": 0.1,
54
  "hidden_size": 1024,
55
  "id2label": {
56
+ "0": "neutral",
57
+ "1": "angry",
58
+ "2": "positive",
59
+ "3": "sad",
60
+ "4": "other"
61
  },
62
  "initializer_range": 0.02,
63
  "intermediate_size": 4096,
64
  "label2id": {
65
+ "neutral": 0,
66
+ "angry": 1,
67
+ "positive": 2,
68
+ "sad": 3,
69
+ "other": 4
70
  },
71
  "layer_norm_eps": 1e-05,
72
  "layerdrop": 0.1,