morit commited on
Commit
3ef7ad4
1 Parent(s): 313e4c0

add label2id mapping

Browse files
Files changed (1) hide show
  1. config.json +3 -3
config.json CHANGED
@@ -20,9 +20,9 @@
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
- "LABEL_0": 0,
24
- "LABEL_1": 1,
25
- "LABEL_2": 2
26
  },
27
  "layer_norm_eps": 1e-05,
28
  "max_position_embeddings": 514,
 
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
+ "entailment": 0,
24
+ "neutral": 1,
25
+ "contradiction": 2
26
  },
27
  "layer_norm_eps": 1e-05,
28
  "max_position_embeddings": 514,