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']

Downloads last month
5
Safetensors
Model size
28.5M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for broadfield-dev/bert-small-ner-pii

Finetuned
(17)
this model
Finetunes
1 model

Dataset used to train broadfield-dev/bert-small-ner-pii

Collections including broadfield-dev/bert-small-ner-pii

Evaluation results