lvwerra HF staff commited on
Commit
b767fcc
1 Parent(s): a46ee84

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -4
config.json CHANGED
@@ -9,13 +9,13 @@
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "neg",
13
- "1": "pos"
14
  },
15
  "initializer_range": 0.02,
16
  "label2id": {
17
- "neg": 0,
18
- "pos": 1
19
  },
20
  "max_position_embeddings": 512,
21
  "model_type": "distilbert",
 
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "NEGATIVE",
13
+ "1": "POSITIVE"
14
  },
15
  "initializer_range": 0.02,
16
  "label2id": {
17
+ "NEGATIVE": 0,
18
+ "POSITIVE": 1
19
  },
20
  "max_position_embeddings": 512,
21
  "model_type": "distilbert",