bert-small-ner-pii-tuned-12261022-tuned-12272248

This model is a fine-tuned version of broadfield-dev/bert-small-ner-pii-tuned-12261022 on the ai4privacy/pii-masking-400k dataset.

Training Details

• Task: TOKEN_CLS
• Columns: Input: source_text Output: privacy_mask
• Rows: Start: 12000 End: 25000
• Epochs: 1
• Learning Rate: 5e-05
• Gradient Accumulation Steps: 8

Entity Labels

['ACCOUNTNUM', 'BUILDINGNUM', 'CITY', 'CREDITCARDNUMBER', 'DATEOFBIRTH', 'DRIVERLICENSENUM', 'EMAIL', 'GIVENNAME', 'IDCARDNUM', 'O', 'PASSWORD', 'SOCIALNUM', 'STREET', 'SURNAME', 'TAXNUM', 'TELEPHONENUM', 'USERNAME', 'ZIPCODE']