Update transformers_recognizer.py

transformers_recognizer.py

@@ -55,7 +55,7 @@ class TransformersRecognizer(EntityRecognizer):
         "ORGANIZATION",
         "AGE",
         "ID",
-        "PHONE",
+        "PHONE_NUMBER",
         "EMAIL",
         "DATE",
         
@@ -71,6 +71,7 @@ class TransformersRecognizer(EntityRecognizer):
         ({"ID"}, {"ID"}),
         ({"EMAIL"}, {"EMAIL"}),
         ({"DATE"}, {"DATE"}),
+        ({"PHONE_NUMBER"}, {"PHONE"}),
         
     ]
 
@@ -80,7 +81,13 @@ class TransformersRecognizer(EntityRecognizer):
         "ORG": "ORGANIZATION",
         "AGE": "AGE",
         "ID": "ID",
-        "EMAIL": "EMAIL"
+        "EMAIL": "EMAIL",
+        "PATIENT": "PERSON",
+        "STAFF": "PERSON",
+        "HOSP": "LOCATION",
+        "PATORG": "ORGANIZATION",
+        "DATE": "DATE_TIME",
+        "PHONE": "PHONE_NUMBER",
     }
 
     DEFAULT_MODEL_PATH = "obi/deid_roberta_i2b2"