Upload XLNetForSequenceClassification

config.json

@@ -15,17 +15,17 @@
   "eos_token_id": 2,
   "ff_activation": "gelu",
   "id2label": {
-    "0": "cardiologist",
-    "1": "traumatologist",
+    "0": "Cardiology",
+    "1": "Traumatology",
     "2": "Mental Health",
-    "3": "pneumologist"
+    "3": "Pneumology"
   },
   "initializer_range": 0.02,
   "label2id": {
+    "Cardiology": 0,
     "Mental Health": 2,
-    "cardiologist": 0,
-    "pneumologist": 3,
-    "traumatologist": 1
+    "Pneumology": 3,
+    "Traumatology": 1
   },
   "layer_norm_eps": 1e-12,
   "mem_len": null,