Hugging Face's logo

Hanish2007
/
brand-safety-model

Text Classification
Transformers
ONNX
Safetensors
distilbert
Generated from Trainer
Inference Endpoints
Model card Files Community
Hanish2007 commited on
Commit
b087cf8
·
verified ·
1 Parent(s): dd84f58

End of training

Browse files
Files changed (3) hide show
  1. README.md +18 -18
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.5795
22
- - Accuracy: 0.8630
23
 
24
  ## Model description
25
 
@@ -38,7 +38,7 @@ More information needed
38
  ### Training hyperparameters
39
 
40
  The following hyperparameters were used during training:
41
- - learning_rate: 3e-05
42
  - train_batch_size: 64
43
  - eval_batch_size: 64
44
  - seed: 42
@@ -50,21 +50,21 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
- | 1.435 | 1.0 | 106 | 1.1905 | 0.7450 |
54
- | 0.8006 | 2.0 | 212 | 0.7220 | 0.8152 |
55
- | 0.5227 | 3.0 | 318 | 0.5708 | 0.8406 |
56
- | 0.4252 | 4.0 | 424 | 0.5011 | 0.8501 |
57
- | 0.3078 | 5.0 | 530 | 0.4905 | 0.8506 |
58
- | 0.2471 | 6.0 | 636 | 0.5174 | 0.8447 |
59
- | 0.1752 | 7.0 | 742 | 0.5095 | 0.8589 |
60
- | 0.2241 | 8.0 | 848 | 0.5265 | 0.8524 |
61
- | 0.1165 | 9.0 | 954 | 0.5525 | 0.8577 |
62
- | 0.102 | 10.0 | 1060 | 0.5510 | 0.8512 |
63
- | 0.0691 | 11.0 | 1166 | 0.5730 | 0.8577 |
64
- | 0.0809 | 12.0 | 1272 | 0.5670 | 0.8619 |
65
- | 0.0631 | 13.0 | 1378 | 0.5715 | 0.8625 |
66
- | 0.0589 | 14.0 | 1484 | 0.5756 | 0.8625 |
67
- | 0.0628 | 15.0 | 1590 | 0.5795 | 0.8630 |
68
 
69
 
70
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.4779
22
+ - Accuracy: 0.8657
23
 
24
  ## Model description
25
 
 
38
  ### Training hyperparameters
39
 
40
  The following hyperparameters were used during training:
41
+ - learning_rate: 2e-05
42
  - train_batch_size: 64
43
  - eval_batch_size: 64
44
  - seed: 42
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | 1.581 | 1.0 | 112 | 1.4547 | 0.6798 |
54
+ | 0.9156 | 2.0 | 224 | 0.8586 | 0.7904 |
55
+ | 0.6402 | 3.0 | 336 | 0.6679 | 0.8292 |
56
+ | 0.5405 | 4.0 | 448 | 0.5595 | 0.8551 |
57
+ | 0.4553 | 5.0 | 560 | 0.5183 | 0.8534 |
58
+ | 0.3509 | 6.0 | 672 | 0.4825 | 0.8629 |
59
+ | 0.316 | 7.0 | 784 | 0.4786 | 0.8584 |
60
+ | 0.2506 | 8.0 | 896 | 0.4710 | 0.8618 |
61
+ | 0.2049 | 9.0 | 1008 | 0.4912 | 0.8567 |
62
+ | 0.1416 | 10.0 | 1120 | 0.4881 | 0.8567 |
63
+ | 0.1572 | 11.0 | 1232 | 0.4779 | 0.8657 |
64
+ | 0.1407 | 12.0 | 1344 | 0.4886 | 0.8596 |
65
+ | 0.132 | 13.0 | 1456 | 0.4933 | 0.8618 |
66
+ | 0.1319 | 14.0 | 1568 | 0.4807 | 0.8635 |
67
+ | 0.1277 | 15.0 | 1680 | 0.4846 | 0.8618 |
68
 
69
 
70
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d07ae015aae003d01f97b5a1a3c9e3dc364765b7cc7eff695ee3035d9cf0ac56
3
  size 267937160
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c77f4e7f86a398915c284a60a1420504d0929a6872b5dd7573df94308ca0814c
3
  size 267937160
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c863f887a899b47469fab8a8080de592e8d9b33244add6965c6e3f9fe01692d9
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ebf0cfba3c64e84f80b252043ebf8463216733324ca62be88caba5c7daba0ee4
3
  size 5240