File size: 1,142 Bytes
9b08c94 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | ---
license: mit
tags:
- deberta
- pii-detection
- token-classification
- kaggle
datasets:
- pii-data-detection
metrics:
- f5-score
---
# DeBERTa-v3-large PII Data Detection (Final Model)
This repository contains the fine-tuned DeBERTa-v3-large model checkpoint from the Kaggle PII Data Detection Competition.
## Model Details
- **Base Architecture**: microsoft/deberta-v3-large
- **Task**: Token classification for Personally Identifiable Information (PII) detection
- **Max Sequence Length**: 4096
- **Training Setup**:
- Competition training data + moredata_dataset_fixed.json
- Optimizer: AdamW, lr=1e-5, weight_decay=0.0
- Batch size: 2
- Epochs: 5 (Fold 0)
- **Validation Metric**: **0.9692** at Epoch 3
## Checkpoints Included
- `deberta_v3_large_w10_ado4_wd0_bs2_lr1e5_5epo_fold0_best.pth` (1.74 GB): Model weights with highest validation score (0.9692).
- `deberta_v3_large_w10_ado4_wd0_bs2_lr1e5_5epo_fold0_last.pth` (5.21 GB): Full training checkpoint including optimizer, scheduler, and scaler states.
- Tokenizer files (`spm.model`, `tokenizer.json`, `config.json`, etc.)
- Training and inference code under `code/`.
|