Downloads · 30 days
4
10% of all-time downloads
bert-small-ner-pii
broadfield-dev/bert-small-ner-pii
bert-small-ner-pii is a machine learning model from broadfield-dev. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This model is a fine-tuned version of prajjwal1/bert-small on the ai4privacy/pii-masking-400k dataset.
All-time downloads
39
Public
Parameters
28.5M
114 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors114 MB · 99%
At a glance
- License
- mit
- Model type
- bert
- Access
- Public
- Created
- Dec 26, 2025
- Updated
- Dec 26, 2025
- SHA
- 33c554f7
Base models
Datasets
Model card
From the Hugging Face model README
bert-small-tuned-12260932
This model is a fine-tuned version of prajjwal1/bert-small on the ai4privacy/pii-masking-400k dataset.
Training Details
- Task: TOKEN_CLS
- Columns: Input: source_text Output: privacy_mask
- Rows: Start: 0 End: 10000
- 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']
- Type
- bert
- License
- mit
- Created
- Dec 26, 2025
- Updated
- Dec 26, 2025