Upload XLNetForSequenceClassification

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "xlnet-large-cased",
+  "_name_or_path": "eperezs/TriageTron-large",
   "architectures": [
     "XLNetForSequenceClassification"
   ],
@@ -15,17 +15,17 @@
   "eos_token_id": 2,
   "ff_activation": "gelu",
   "id2label": {
-    "0": "Mental Health",
-    "1": "cardiologist",
-    "2": "pneumologist",
-    "3": "traumatologist"
+    "0": "cardiologist",
+    "1": "traumatologist",
+    "2": "Mental Health",
+    "3": "pneumologist"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "Mental Health": 0,
-    "cardiologist": 1,
-    "pneumologist": 2,
-    "traumatologist": 3
+    "Mental Health": 2,
+    "cardiologist": 0,
+    "pneumologist": 3,
+    "traumatologist": 1
   },
   "layer_norm_eps": 1e-12,
   "mem_len": null,