Downloads · 30 days
16
34% of all-time downloads
pii-ner-model-gretel
sharanharsoor/pii-ner-model-gretel
pii-ner-model-gretel is a machine learning model from sharanharsoor. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model is trained to detect Personal Identifiable Information (PII) in text. It can identify various types of PII including:
All-time downloads
47
Public
Parameters
66.4M
531 MB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors266 MB · 100%
At a glance
- Model type
- distilbert
- Access
- Public
- Created
- Jan 14, 2025
- Updated
- Jan 14, 2025
- SHA
- 342471e9
Model card
From the Hugging Face model README
PII NER Model
This model is trained to detect Personal Identifiable Information (PII) in text. It can identify various types of PII including:
- Names (first and last)
- Email addresses
- Phone numbers
- Social Security Numbers (SSN)
- Credit card numbers
- Medical record numbers
- Employee IDs
- Addresses
- And more
Usage
from transformers import AutoModelForTokenClassification, AutoTokenizer
model = AutoModelForTokenClassification.from_pretrained("sharanharsoor/pii-ner-model")
tokenizer = AutoTokenizer.from_pretrained("sharanharsoor/pii-ner-model")
Training Data
The model was trained on the Gretel PII dataset with custom modifications.
Performance
The model achieves:
- F1 Score: 0.9451
- Precision: 0.9401
- Recall: 0.9502
- Accuracy: 0.9952
- Type
- distilbert
- Created
- Jan 14, 2025
- Updated
- Jan 14, 2025