alex-miller commited on
Commit
0f72810
1 Parent(s): b53e992

End of training

Browse files
Files changed (2) hide show
  1. README.md +17 -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 [alex-miller/ODABert](https://huggingface.co/alex-miller/ODABert) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.7126
24
- - Accuracy: 0.8881
25
- - F1: 0.8100
26
- - Precision: 0.7556
27
- - Recall: 0.8728
28
 
29
  ## Model description
30
 
@@ -49,22 +49,27 @@ The following hyperparameters were used during training:
49
  - seed: 42
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: linear
52
- - num_epochs: 5
53
 
54
  ### Training results
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
57
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
58
- | 0.962 | 1.0 | 437 | 0.9254 | 0.8485 | 0.7272 | 0.7161 | 0.7386 |
59
- | 0.8959 | 2.0 | 874 | 0.8276 | 0.8762 | 0.7906 | 0.7354 | 0.8546 |
60
- | 0.8419 | 3.0 | 1311 | 0.7560 | 0.8801 | 0.7925 | 0.7517 | 0.8379 |
61
- | 0.7452 | 4.0 | 1748 | 0.7235 | 0.8856 | 0.8048 | 0.7540 | 0.8630 |
62
- | 0.7234 | 5.0 | 2185 | 0.7126 | 0.8881 | 0.8100 | 0.7556 | 0.8728 |
 
 
 
 
 
63
 
64
 
65
  ### Framework versions
66
 
67
  - Transformers 4.38.2
68
- - Pytorch 2.2.1+cu121
69
  - Datasets 2.18.0
70
  - Tokenizers 0.15.2
 
20
 
21
  This model is a fine-tuned version of [alex-miller/ODABert](https://huggingface.co/alex-miller/ODABert) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.6084
24
+ - Accuracy: 0.9116
25
+ - F1: 0.8471
26
+ - Precision: 0.8029
27
+ - Recall: 0.8966
28
 
29
  ## Model description
30
 
 
49
  - seed: 42
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: linear
52
+ - num_epochs: 10
53
 
54
  ### Training results
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
57
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
58
+ | 0.9604 | 1.0 | 437 | 0.9473 | 0.8617 | 0.7483 | 0.7446 | 0.7519 |
59
+ | 0.8897 | 2.0 | 874 | 0.8169 | 0.8804 | 0.7995 | 0.7380 | 0.8721 |
60
+ | 0.8054 | 3.0 | 1311 | 0.7170 | 0.8858 | 0.8028 | 0.7601 | 0.8505 |
61
+ | 0.6757 | 4.0 | 1748 | 0.6679 | 0.8921 | 0.8164 | 0.7631 | 0.8777 |
62
+ | 0.6153 | 5.0 | 2185 | 0.6279 | 0.8992 | 0.8275 | 0.7772 | 0.8847 |
63
+ | 0.5782 | 6.0 | 2622 | 0.5773 | 0.8995 | 0.8305 | 0.7705 | 0.9008 |
64
+ | 0.5435 | 7.0 | 3059 | 0.6106 | 0.9072 | 0.8401 | 0.7937 | 0.8924 |
65
+ | 0.5333 | 8.0 | 3496 | 0.6141 | 0.9079 | 0.8435 | 0.7878 | 0.9078 |
66
+ | 0.5349 | 9.0 | 3933 | 0.6056 | 0.9097 | 0.8448 | 0.7964 | 0.8994 |
67
+ | 0.539 | 10.0 | 4370 | 0.6084 | 0.9116 | 0.8471 | 0.8029 | 0.8966 |
68
 
69
 
70
  ### Framework versions
71
 
72
  - Transformers 4.38.2
73
+ - Pytorch 2.4.1+cu121
74
  - Datasets 2.18.0
75
  - Tokenizers 0.15.2
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d9f0f8830185e0b111e5ab43248664d4e81e92df9804ff8047113f6d0d0ef125
3
  size 672720896
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5df8050b854c995b00bc837947ff4c7039e3ea48293ab2a43f31477d90532d9c
3
  size 672720896