PII NER Fine-tuned DistilBERT
Overview
Fine-tuned DistilBERT model for detecting Personally Identifiable Information (PII).
Use Cases
- PII detection
- Data redaction
- Privacy filtering
Model Type
- Token Classification (NER)
- PyTorch fine-tuned model
Training Dataset
English AI4Privacy Dataset
The model was fine-tuned using the English AI4Privacy dataset containing synthetic text annotated with privacy-related entities. https://www.kaggle.com/datasets/verracodeguacas/ai4privacy-pii
Usage
from transformers import pipeline
pipe = pipeline( "token-classification", model="samuelolubukun/pii-ner-finetuned-distilbert" )
- Downloads last month
- 27