metadata
license: mit
tags:
- token_cls
- generated_from_trainer
base_model: broadfield-dev/bert-mini-tuned-12261137-tuned-12261248
datasets:
- ai4privacy/pii-masking-400k
model-index:
- name: bert-mini-tuned-12261137-tuned-12261248-tuned-12261952
results: []
bert-mini-tuned-12261137-tuned-12261248-tuned-12261952
This model is a fine-tuned version of broadfield-dev/bert-mini-tuned-12261137-tuned-12261248 on the ai4privacy/pii-masking-400k dataset.
Training Details
- Task: TOKEN_CLS
- Columns: Input: source_text Output: privacy_mask
- Rows: Start: 90000 End: 150000
- Epochs: 4
- Learning Rate: 2e-05
- Gradient Accumulation Steps: 4
Entity Labels
['ACCOUNTNUM', 'BUILDINGNUM', 'CITY', 'CREDITCARDNUMBER', 'DATEOFBIRTH', 'DRIVERLICENSENUM', 'EMAIL', 'GIVENNAME', 'IDCARDNUM', 'O', 'PASSWORD', 'SOCIALNUM', 'STREET', 'SURNAME', 'TAXNUM', 'TELEPHONENUM', 'USERNAME', 'ZIPCODE']