digitalhealth-healthyliving
/

MediFlow

Text Classification

Inference Endpoints

Model card Files Files and versions Community

eperezs commited on Jul 15

Commit

a1b0e5d

•

1 Parent(s): 0bf12fb

Upload XLNetForSequenceClassification

Files changed (2) hide show

config.json +8 -8
model.safetensors +1 -1

config.json CHANGED Viewed

@@ -15,17 +15,17 @@
   "eos_token_id": 2,
   "ff_activation": "gelu",
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2,
-    "LABEL_3": 3
   },
   "layer_norm_eps": 1e-12,
   "mem_len": null,

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

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:17b6e6b47b7106c3288575eec2382e01854e8d4d5def9df8302b79c56e55b2e9
 size 1445335712

 version https://git-lfs.github.com/spec/v1
+oid sha256:3502369bf4df7427ddfea5b6c91de88da4bff12b4b708994e6f90c9f7f7998f5
 size 1445335712