Upload RobertaForTokenClassification

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "./fine_tuned_deid_roberta_i2b2",
+  "_name_or_path": "./trained_clinicalbert_model",
   "architectures": [
     "RobertaForTokenClassification"
   ],
@@ -12,100 +12,44 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 1024,
   "id2label": {
-    "0": "B-AGE",
-    "1": "B-DATE",
-    "2": "B-EMAIL",
-    "3": "B-HOSP",
-    "4": "B-ID",
-    "5": "B-LOC",
-    "6": "B-OTHERPHI",
-    "7": "B-PATIENT",
-    "8": "B-PATORG",
-    "9": "B-PHONE",
-    "10": "B-STAFF",
-    "11": "I-AGE",
-    "12": "I-DATE",
-    "13": "I-EMAIL",
-    "14": "I-HOSP",
-    "15": "I-ID",
-    "16": "I-LOC",
-    "17": "I-OTHERPHI",
-    "18": "I-PATIENT",
-    "19": "I-PATORG",
-    "20": "I-PHONE",
-    "21": "I-STAFF",
-    "22": "L-AGE",
-    "23": "L-DATE",
-    "24": "L-EMAIL",
-    "25": "L-HOSP",
-    "26": "L-ID",
-    "27": "L-LOC",
-    "28": "L-OTHERPHI",
-    "29": "L-PATIENT",
-    "30": "L-PATORG",
-    "31": "L-PHONE",
-    "32": "L-STAFF",
-    "33": "O",
-    "34": "U-AGE",
-    "35": "U-DATE",
-    "36": "U-EMAIL",
-    "37": "U-HOSP",
-    "38": "U-ID",
-    "39": "U-LOC",
-    "40": "U-OTHERPHI",
-    "41": "U-PATIENT",
-    "42": "U-PATORG",
-    "43": "U-PHONE",
-    "44": "U-STAFF"
+    "0": "O",
+    "1": "B-DOCTOR_NAME",
+    "2": "B-HOSPITAL_NAME",
+    "3": "B-PATIENT_NAME",
+    "4": "B-PATIENT_ID",
+    "5": "B-OTHER_NAME",
+    "6": "B-PHONE_NUMBER",
+    "7": "B-BIRTHDAY",
+    "8": "B-LOCATION",
+    "9": "I-DOCTOR_NAME",
+    "10": "I-HOSPITAL_NAME",
+    "11": "I-PATIENT_NAME",
+    "12": "I-PATIENT_ID",
+    "13": "I-OTHER_NAME",
+    "14": "I-PHONE_NUMBER",
+    "15": "I-BIRTHDAY",
+    "16": "I-LOCATION"
   },
   "initializer_range": 0.02,
   "intermediate_size": 4096,
   "label2id": {
-    "B-AGE": 0,
-    "B-DATE": 1,
-    "B-EMAIL": 2,
-    "B-HOSP": 3,
-    "B-ID": 4,
-    "B-LOC": 5,
-    "B-OTHERPHI": 6,
-    "B-PATIENT": 7,
-    "B-PATORG": 8,
-    "B-PHONE": 9,
-    "B-STAFF": 10,
-    "I-AGE": 11,
-    "I-DATE": 12,
-    "I-EMAIL": 13,
-    "I-HOSP": 14,
-    "I-ID": 15,
-    "I-LOC": 16,
-    "I-OTHERPHI": 17,
-    "I-PATIENT": 18,
-    "I-PATORG": 19,
-    "I-PHONE": 20,
-    "I-STAFF": 21,
-    "L-AGE": 22,
-    "L-DATE": 23,
-    "L-EMAIL": 24,
-    "L-HOSP": 25,
-    "L-ID": 26,
-    "L-LOC": 27,
-    "L-OTHERPHI": 28,
-    "L-PATIENT": 29,
-    "L-PATORG": 30,
-    "L-PHONE": 31,
-    "L-STAFF": 32,
-    "O": 33,
-    "U-AGE": 34,
-    "U-DATE": 35,
-    "U-EMAIL": 36,
-    "U-HOSP": 37,
-    "U-ID": 38,
-    "U-LOC": 39,
-    "U-OTHERPHI": 40,
-    "U-PATIENT": 41,
-    "U-PATORG": 42,
-    "U-PHONE": 43,
-    "U-STAFF": 44
+    "B-BIRTHDAY": 7,
+    "B-DOCTOR_NAME": 1,
+    "B-HOSPITAL_NAME": 2,
+    "B-LOCATION": 8,
+    "B-OTHER_NAME": 5,
+    "B-PATIENT_ID": 4,
+    "B-PATIENT_NAME": 3,
+    "B-PHONE_NUMBER": 6,
+    "I-BIRTHDAY": 15,
+    "I-DOCTOR_NAME": 9,
+    "I-HOSPITAL_NAME": 10,
+    "I-LOCATION": 16,
+    "I-OTHER_NAME": 13,
+    "I-PATIENT_ID": 12,
+    "I-PATIENT_NAME": 11,
+    "I-PHONE_NUMBER": 14,
+    "O": 0
   },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0fe31b811744cd3cf69d7a5a4daef4143cbe57b37ae4b2d76bab1c055c9e079f
-size 1417473100
+oid sha256:387053a2c52475b2a9bcc0648ac2fec6e25a3e8cdd5cd1924d249cce802bce34
+size 1417358292