David Chu commited on
Commit
625d89b
1 Parent(s): cbe46fb

fix: add label2id and id2label

Browse files
Files changed (1) hide show
  1. config.json +11 -1
config.json CHANGED
@@ -18,5 +18,15 @@
18
  "post_attention_groups": 1,
19
  "intermediate_groups": 4,
20
  "output_groups": 4,
21
- "num_labels": 3
 
 
 
 
 
 
 
 
 
 
22
  }
18
  "post_attention_groups": 1,
19
  "intermediate_groups": 4,
20
  "output_groups": 4,
21
+ "num_labels": 3,
22
+ "label2id": {
23
+ "ENTAILMENT": 0,
24
+ "NEUTRAL": 1,
25
+ "CONTRADICTION": 2,
26
+ },
27
+ "id2label": {
28
+ "0": "ENTAILMENT",
29
+ "1": "NEUTRAL",
30
+ "2": "CONTRADICTION"
31
+ }
32
  }