nlpie
/

clinical-mobilebert-i2b2-2010

Token Classification

Model card Files Files and versions

mohammadmahdinouri commited on Apr 20, 2023

Commit

c02a0f0

·

1 Parent(s): b0414b3

Update config.json

Files changed (1) hide show

config.json +7 -7

config.json CHANGED Viewed

@@ -24,13 +24,13 @@
   "intra_bottleneck_size": 128,
   "key_query_shared_bottleneck": true,
   "label2id": {
-    "LABEL_1": 0,
-    "LABEL_2": 1,
-    "LABEL_3": 2,
-    "LABEL_4": 3,
-    "LABEL_5": 4,
-    "LABEL_6": 5,
-    "LABEL_7": 6
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,

   "intra_bottleneck_size": 128,
   "key_query_shared_bottleneck": true,
   "label2id": {
+    "I-treatment": 0,
+    "O": 1,
+    "B-test": 2,
+    "I-problem": 3,
+    "B-treatment": 4,
+    "I-test": 5,
+    "B-problem": 6
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,