Align label mapping with wnli config of glue dataset
#1
by
lewtun
HF staff
- opened
- config.json +10 -2
config.json
CHANGED
@@ -40,5 +40,13 @@
|
|
40 |
"untie_r": true,
|
41 |
"use_mems_eval": true,
|
42 |
"use_mems_train": false,
|
43 |
-
"vocab_size": 32000
|
44 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
40 |
"untie_r": true,
|
41 |
"use_mems_eval": true,
|
42 |
"use_mems_train": false,
|
43 |
+
"vocab_size": 32000,
|
44 |
+
"label2id": {
|
45 |
+
"not_entailment": 0,
|
46 |
+
"entailment": 1
|
47 |
+
},
|
48 |
+
"id2label": {
|
49 |
+
"0": "not_entailment",
|
50 |
+
"1": "entailment"
|
51 |
+
}
|
52 |
+
}
|