Hugging Face's logo

PuxAI
/
PII-Binary-Filter-Extreme-Recall-Fix

Text Classification
Transformers
Safetensors
deberta-v2
Generated from Trainer
text-embeddings-inference
Model card Files
xet
 Community
PuxAI commited on
Commit
9b6adf4
·
verified ·
1 Parent(s): 28ca7fe

Completed Filter Training With Full TP/FP/FN Distribution

Browse files
Files changed (2) hide show
  1. README.md +12 -12
  2. model.safetensors +1 -1
README.md CHANGED
@@ -20,11 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [microsoft/deberta-v3-small](https://huggingface.co/microsoft/deberta-v3-small) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.1328
24
- - F1: 0.9811
25
- - Recall: 0.9969
26
- - Precision: 0.9658
27
- - Trash Caught: 0.2798
28
 
29
  ## Model description
30
 
@@ -43,8 +43,8 @@ More information needed
43
  ### Training hyperparameters
44
 
45
  The following hyperparameters were used during training:
46
- - learning_rate: 1e-05
47
- - train_batch_size: 32
48
  - eval_batch_size: 64
49
  - seed: 42
50
  - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
@@ -55,11 +55,11 @@ The following hyperparameters were used during training:
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | F1 | Recall | Precision | Trash Caught |
57
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:---------:|:------------:|
58
- | No log | 1.0 | 250 | 0.1488 | 0.9767 | 0.9978 | 0.9565 | 0.0734 |
59
- | 0.3046 | 2.0 | 500 | 0.1534 | 0.9774 | 0.9996 | 0.9562 | 0.0642 |
60
- | 0.3046 | 3.0 | 750 | 0.1259 | 0.9800 | 0.9969 | 0.9638 | 0.2339 |
61
- | 0.1678 | 4.0 | 1000 | 0.1344 | 0.9806 | 0.9973 | 0.9644 | 0.2477 |
62
- | 0.1678 | 5.0 | 1250 | 0.1328 | 0.9811 | 0.9969 | 0.9658 | 0.2798 |
63
 
64
 
65
  ### Framework versions
 
20
 
21
  This model is a fine-tuned version of [microsoft/deberta-v3-small](https://huggingface.co/microsoft/deberta-v3-small) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.1361
24
+ - F1: 0.9852
25
+ - Recall: 0.9879
26
+ - Precision: 0.9826
27
+ - Trash Caught: 0.6422
28
 
29
  ## Model description
30
 
 
43
  ### Training hyperparameters
44
 
45
  The following hyperparameters were used during training:
46
+ - learning_rate: 3e-05
47
+ - train_batch_size: 16
48
  - eval_batch_size: 64
49
  - seed: 42
50
  - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | F1 | Recall | Precision | Trash Caught |
57
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:---------:|:------------:|
58
+ | No log | 1.0 | 499 | 0.1403 | 0.9785 | 0.9953 | 0.9622 | 0.2018 |
59
+ | 0.3066 | 2.0 | 998 | 0.1123 | 0.9832 | 0.9908 | 0.9757 | 0.4954 |
60
+ | 0.179 | 3.0 | 1497 | 0.1188 | 0.9853 | 0.9910 | 0.9796 | 0.5780 |
61
+ | 0.1209 | 4.0 | 1996 | 0.1293 | 0.9857 | 0.9921 | 0.9794 | 0.5734 |
62
+ | 0.0834 | 5.0 | 2495 | 0.1361 | 0.9852 | 0.9879 | 0.9826 | 0.6422 |
63
 
64
 
65
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d587665109aee16a6527d60be5106c0ce75dc50ba22f26764c6cadec27a04a7b
3
  size 567300568
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b1619c016404d0ab2979857745cd4e1a46f7827de6070ccc7f87f3374d51791
3
  size 567300568