PII-Filter-SpanBased-Stage2
This model is a fine-tuned version of microsoft/deberta-v3-small on the None dataset. It achieves the following results on the evaluation set:
- Loss: 0.0638
- F1: 0.9930
- Recall: 0.9928
- Precision: 0.9933
- Trash Caught: 0.7130
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 3e-05
- train_batch_size: 16
- eval_batch_size: 64
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- num_epochs: 4
Training results
| Training Loss | Epoch | Step | Validation Loss | F1 | Recall | Precision | Trash Caught |
|---|---|---|---|---|---|---|---|
| 0.3289 | 1.0 | 570 | 0.0704 | 0.9909 | 0.9978 | 0.9840 | 0.3108 |
| 0.2035 | 2.0 | 1140 | 0.0604 | 0.9908 | 0.9915 | 0.9900 | 0.5740 |
| 0.1326 | 3.0 | 1710 | 0.0638 | 0.9921 | 0.9903 | 0.9938 | 0.7386 |
| 0.0935 | 4.0 | 2280 | 0.0638 | 0.9930 | 0.9928 | 0.9933 | 0.7130 |
Framework versions
- Transformers 4.56.0
- Pytorch 2.8.0+cu129
- Datasets 4.8.3
- Tokenizers 0.22.0
- Downloads last month
- 95
Model tree for PuxAI/PII-Filter-SpanBased-Stage2
Base model
microsoft/deberta-v3-small