changing is_regression to unified API

2019-06-26 09:54:05 +02:00
parent e55d4c4ede
commit 092dacfd62
4 changed files with 49 additions and 31 deletions
--- a/examples/utils_glue.py
+++ b/examples/utils_glue.py
@@ -591,3 +591,15 @@ output_modes = {
    "rte": "classification",
    "wnli": "classification",
 }
+
+GLUE_TASKS_NUM_LABELS = {
+    "cola": 2,
+    "mnli": 3,
+    "mrpc": 2,
+    "sst-2": 2,
+    "sts-b": 1,
+    "qqp": 2,
+    "qnli": 2,
+    "rte": 2,
+    "wnli": 2,
+}