Dc-4nderson commited on
Commit
439b164
·
verified ·
1 Parent(s): dc4329c

set id to label

Browse files
Files changed (1) hide show
  1. config.json +16 -16
config.json CHANGED
@@ -9,24 +9,24 @@
9
  "dtype": "float32",
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": 0,
13
- "1": 1,
14
- "2": 2,
15
- "3": 3,
16
- "4": 4,
17
- "5": 5,
18
- "6": 6
19
- },
20
  "initializer_range": 0.02,
21
  "label2id": {
22
- "0": 0,
23
- "1": 1,
24
- "2": 2,
25
- "3": 3,
26
- "4": 4,
27
- "5": 5,
28
- "6": 6
29
- },
30
  "max_position_embeddings": 512,
31
  "model_type": "distilbert",
32
  "n_heads": 12,
 
9
  "dtype": "float32",
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "informative",
13
+ "1": "motivational",
14
+ "2": "optimistic",
15
+ "3": "practical",
16
+ "4": "reflective",
17
+ "5": "thoughtful",
18
+ "6": "uplifting"
19
+ },
20
  "initializer_range": 0.02,
21
  "label2id": {
22
+ "informative": 0,
23
+ "motivational": 1,
24
+ "optimistic": 2,
25
+ "practical": 3,
26
+ "reflective": 4,
27
+ "thoughtful": 5,
28
+ "uplifting": 6
29
+ },
30
  "max_position_embeddings": 512,
31
  "model_type": "distilbert",
32
  "n_heads": 12,