beki commited on
Commit
b772d2f
·
1 Parent(s): 0d6a3f0

Update flair_recognizer.py

Browse files
Files changed (1) hide show
  1. flair_recognizer.py +5 -8
flair_recognizer.py CHANGED
@@ -78,17 +78,13 @@ class FlairRecognizer(EntityRecognizer):
78
  ({"AGE"}, {"AGE"}),
79
  ({"CURRENCY"}, {"CURRENCY"}),
80
  ({"SWIFT_CODE"}, {"SWIFT_CODE"}),
81
-
82
- # ({"US_ITIN"}, {"US_ITIN"}),
83
- # ({"US_BANK_NUMBER"}, {"US_BANK_NUMBER"}),
84
- # ({"US_DRIVER_LICENSE"}, {"US_DRIVER_LICENSE"}),
85
  ]
86
 
87
  MODEL_LANGUAGES = {
88
  "en": "beki/flair-ner-debug-english",
89
- # "es": "flair/ner-spanish-large",
90
- # "de": "flair/ner-german-large",
91
- # "nl": "flair/ner-dutch-large",
92
  }
93
 
94
  PRESIDIO_EQUIVALENCES = {
@@ -102,7 +98,8 @@ class FlairRecognizer(EntityRecognizer):
102
  "IBAN_CODE": "IBAN_CODE",
103
  "IP_ADDRESS": "IP_ADDRESS",
104
  "EMAIL_ADDRESS": "EMAIL",
105
- # 'MISC': 'MISCELLANEOUS' # - Probably not PII
 
106
  }
107
 
108
  def __init__(
 
78
  ({"AGE"}, {"AGE"}),
79
  ({"CURRENCY"}, {"CURRENCY"}),
80
  ({"SWIFT_CODE"}, {"SWIFT_CODE"}),
81
+ ({"US_ITIN"}, {"US_ITIN"}),
82
+ ({"US_BANK_NUMBER"}, {"US_BANK_NUMBER"}),
83
+ ({"US_DRIVER_LICENSE"}, {"US_DRIVER_LICENSE"}),
 
84
  ]
85
 
86
  MODEL_LANGUAGES = {
87
  "en": "beki/flair-ner-debug-english",
 
 
 
88
  }
89
 
90
  PRESIDIO_EQUIVALENCES = {
 
98
  "IBAN_CODE": "IBAN_CODE",
99
  "IP_ADDRESS": "IP_ADDRESS",
100
  "EMAIL_ADDRESS": "EMAIL",
101
+ "US_DRIVER_LICENSE": "US_DRIVER_LICENSE",
102
+ "US_BANK_NUMBER": "US_BANK_NUMBER",
103
  }
104
 
105
  def __init__(