Align label mapping with rte config of glue dataset
Browse filesHi there, your model is using a default label mapping. Accept this PR to align the label mapping with the `rte` config of the `glue` dataset this model was trained on. This will enable your model to be evaluated by [Hugging Face's automatic model evaluator](https://huggingface.co/spaces/autoevaluate/model-evaluator?dataset=glue)
- 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 |
+
"entailment": 0,
|
46 |
+
"not_entailment": 1
|
47 |
+
},
|
48 |
+
"id2label": {
|
49 |
+
"0": "entailment",
|
50 |
+
"1": "not_entailment"
|
51 |
+
}
|
52 |
+
}
|