Ramamurthi commited on
Commit
8d61a4c
1 Parent(s): 3e09a6d

Update config.json

Browse files

Updating the labels of config.json so that the labels are displayed properly for anyone who uses the inference API

Files changed (1) hide show
  1. config.json +10 -10
config.json CHANGED
@@ -9,19 +9,19 @@
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1",
14
- "2": "LABEL_2",
15
- "3": "LABEL_3",
16
- "4": "LABEL_4"
17
  },
18
  "initializer_range": 0.02,
19
  "label2id": {
20
- "LABEL_0": 0,
21
- "LABEL_1": 1,
22
- "LABEL_2": 2,
23
- "LABEL_3": 3,
24
- "LABEL_4": 4
25
  },
26
  "max_position_embeddings": 512,
27
  "model_type": "distilbert",
 
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "1 star",
13
+ "1": "2 star",
14
+ "2": "3 star",
15
+ "3": "4 star",
16
+ "4": "5 star"
17
  },
18
  "initializer_range": 0.02,
19
  "label2id": {
20
+ "1 star": 0,
21
+ "2 star": 1,
22
+ "3 star": 2,
23
+ "4 star": 3,
24
+ "5 star": 4
25
  },
26
  "max_position_embeddings": 512,
27
  "model_type": "distilbert",