zcamz commited on
Commit
4f6649d
1 Parent(s): 79927b7

zcamz/bert-classifier-toxic

Browse files
README.md CHANGED
@@ -18,10 +18,10 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0492
22
- - F1: 0.5175
23
- - Roc Auc: 0.7393
24
- - Accuracy: 0.9206
25
 
26
  ## Model description
27
 
@@ -46,13 +46,22 @@ The following hyperparameters were used during training:
46
  - seed: 42
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
49
- - num_epochs: 1
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|:--------:|
55
- | No log | 1.0 | 499 | 0.0492 | 0.5175 | 0.7393 | 0.9206 |
 
 
 
 
 
 
 
 
 
56
 
57
 
58
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.0409
22
+ - F1: 0.7159
23
+ - Roc Auc: 0.8423
24
+ - Accuracy: 0.9246
25
 
26
  ## Model description
27
 
 
46
  - seed: 42
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
49
+ - num_epochs: 10
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|:--------:|
55
+ | No log | 1.0 | 499 | 0.0442 | 0.5351 | 0.7417 | 0.9219 |
56
+ | 0.09 | 2.0 | 998 | 0.0395 | 0.6956 | 0.8168 | 0.9272 |
57
+ | 0.04 | 3.0 | 1497 | 0.0396 | 0.7040 | 0.8348 | 0.9254 |
58
+ | 0.0332 | 4.0 | 1996 | 0.0409 | 0.7159 | 0.8423 | 0.9246 |
59
+ | 0.0287 | 5.0 | 2495 | 0.0431 | 0.7013 | 0.8188 | 0.9278 |
60
+ | 0.0247 | 6.0 | 2994 | 0.0452 | 0.7146 | 0.8392 | 0.9229 |
61
+ | 0.0215 | 7.0 | 3493 | 0.0469 | 0.7094 | 0.8335 | 0.9256 |
62
+ | 0.0189 | 8.0 | 3992 | 0.0497 | 0.7046 | 0.8363 | 0.9205 |
63
+ | 0.0166 | 9.0 | 4491 | 0.0512 | 0.7135 | 0.8464 | 0.9202 |
64
+ | 0.0153 | 10.0 | 4990 | 0.0515 | 0.7125 | 0.8426 | 0.9214 |
65
 
66
 
67
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:708b3b5b367064824510dfb97d4dba5a2070b5f74ebcdcead5389ce06cb49eeb
3
  size 437970952
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:697417264090c9a4a5e6c58602af99e5e825c93c5e7c89b62f83a89c53aedd31
3
  size 437970952
runs/May18_22-01-16_0db05e2c36fe/events.out.tfevents.1716069696.0db05e2c36fe.2391.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d1735fb53f6513afdad12203de00fbc35e6716028e7ab3331c172ef6b75008c
3
+ size 5008
runs/May18_22-02-17_0db05e2c36fe/events.out.tfevents.1716069745.0db05e2c36fe.2391.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82d7cf71151482dfa2ff56827f41f015e930cf941ae01e7dca7ffd9c93a12d57
3
+ size 5007
runs/May18_22-03-04_0db05e2c36fe/events.out.tfevents.1716069790.0db05e2c36fe.2391.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4db6eded742417a9af2375b0e0f9c72f8148528cae617abcd168a46ae5a9512
3
+ size 5008
runs/May18_22-03-37_0db05e2c36fe/events.out.tfevents.1716069822.0db05e2c36fe.2391.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:149c2eab0bc970f9f0ee9cbd57f8a29f6f955f61603b6b7686d171a5ab8af919
3
+ size 11461
runs/May18_22-03-37_0db05e2c36fe/events.out.tfevents.1716076039.0db05e2c36fe.2391.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1508a076faa2787b33e58ac8fa08df208daadca09ed7fda8049c905e6ce93ca
3
+ size 508
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d14230756697097c404fe67b17f12ac944b13f0b26042dde27c84aadecdc6bb6
3
  size 4984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c9d9f57440bb98e9aafbcaeb0a563cfafd187c0f42f680fb42d8dab45e62867
3
  size 4984