Downloads · 30 days
44
36% of all-time downloads
pii-distilbert-en
munibz/pii-distilbert-en
pii-distilbert-en is a token classification model from munibz. Use it when you need labels on individual words, such as names. The card lists the license as apache-2.0.
Fine-tuned distilbert-base-cased for PII (Personally Identifiable Information) detection on the English subset of ai4privacy/pii-masking-300k.
All-time downloads
123
Public
Parameters
65.2M
261 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors261 MB · 100%
At a glance
- Task
- Token Classification
- License
- apache-2.0
- Model type
- distilbert
- Access
- Public
- Created
- Apr 30, 2026
- Updated
- Apr 30, 2026
- SHA
- 5ffd58e7
Try a prompt
Base models
Datasets
Model card
From the Hugging Face model README
PII Detection — DistilBERT (English)
Fine-tuned distilbert-base-cased for PII (Personally Identifiable Information) detection
on the English subset of ai4privacy/pii-masking-300k.
Results
- F1: 0.96
- 30k training examples, 5k validation, 6 epochs on a Colab T4
Entity types
PERSON, EMAIL, PHONE, USERNAME, ID_NUMBER, ADDRESS, IP_ADDRESS,
URL, DATE_TIME, PASSWORD, DEMOGRAPHIC, OTHER
Usage
from transformers import pipeline
ner = pipeline("ner", model="munibz/pii-distilbert-en", aggregation_strategy="simple")
text = "Hi, I'm John Smith. Email me at [email protected]."
print(ner(text))
Training
- Base model: distilbert-base-cased
- Max length: 384
- Batch size: 32
- Learning rate: 3e-5 (cosine schedule, 10% warmup)
- Epochs: 6 (early stopping patience 2)
- FP16 on T4 GPU
- Task
- Token Classification
- Type
- distilbert
- License
- apache-2.0
- Languages
- en
- Created
- Apr 30, 2026
- Updated
- Apr 30, 2026