broadfield-dev's picture
Upload README.md with huggingface_hub
d04af66 verified
---
license: mit
tags:
- token_cls
- generated_from_trainer
base_model: broadfield-dev/bert-tiny-training-mid
datasets:
- ai4privacy/pii-masking-400k
model-index:
- name: bert-tiny-training-mid-tuned-12260542
results: []
---
# bert-tiny-training-mid-tuned-12260542
This model is a fine-tuned version of [broadfield-dev/bert-tiny-training-mid](https://huggingface.co/broadfield-dev/bert-tiny-training-mid) on the [ai4privacy/pii-masking-400k](https://huggingface.co/ai4privacy/pii-masking-400k) dataset.
## Training Details
- **Task:** TOKEN_CLS
- **Columns:** Input: source_text Output: privacy_mask
- **Rows:** Start: 170000 End: 240000
- **Epochs:** 1
- **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']`