| license: mit | |
| tags: | |
| - token_cls | |
| - generated_from_trainer | |
| base_model: prajjwal1/bert-tiny | |
| datasets: | |
| - ai4privacy/pii-masking-400k | |
| model-index: | |
| - name: bert-tiny-tuned-12260146 | |
| results: [] | |
| # bert-tiny-tuned-12260146 | |
| This model is a fine-tuned version of [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) 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: 0 End: 20000 | |
| - **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']` | |