alex-miller commited on
Commit
445b1de
1 Parent(s): 944bdf0

End of training

Browse files
README.md ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: alex-miller/ODABert
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ - f1
9
+ - precision
10
+ - recall
11
+ model-index:
12
+ - name: iati-gender-multi-classifier-weighted-minitest
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # iati-gender-multi-classifier-weighted-minitest
20
+
21
+ This model is a fine-tuned version of [alex-miller/ODABert](https://huggingface.co/alex-miller/ODABert) on the None dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.6625
24
+ - Accuracy: 0.8256
25
+ - F1: 0.6997
26
+ - Precision: 0.6138
27
+ - Recall: 0.8135
28
+
29
+ ## Model description
30
+
31
+ More information needed
32
+
33
+ ## Intended uses & limitations
34
+
35
+ More information needed
36
+
37
+ ## Training and evaluation data
38
+
39
+ More information needed
40
+
41
+ ## Training procedure
42
+
43
+ ### Training hyperparameters
44
+
45
+ The following hyperparameters were used during training:
46
+ - learning_rate: 2e-06
47
+ - train_batch_size: 16
48
+ - eval_batch_size: 16
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.9142 | 1.0 | 616 | 0.7573 | 0.7352 | 0.5803 | 0.4802 | 0.7331 |
59
+ | 0.7431 | 2.0 | 1232 | 0.6871 | 0.7973 | 0.6652 | 0.5661 | 0.8062 |
60
+ | 0.6934 | 3.0 | 1848 | 0.6713 | 0.8160 | 0.6902 | 0.5955 | 0.8208 |
61
+ | 0.6631 | 4.0 | 2464 | 0.6609 | 0.8187 | 0.6944 | 0.5997 | 0.8245 |
62
+ | 0.6341 | 5.0 | 3080 | 0.6589 | 0.8237 | 0.6956 | 0.6117 | 0.8062 |
63
+ | 0.6143 | 6.0 | 3696 | 0.6551 | 0.8205 | 0.6970 | 0.6027 | 0.8263 |
64
+ | 0.5931 | 7.0 | 4312 | 0.6597 | 0.8201 | 0.6893 | 0.6061 | 0.7989 |
65
+ | 0.5773 | 8.0 | 4928 | 0.6579 | 0.8224 | 0.6963 | 0.6076 | 0.8154 |
66
+ | 0.5652 | 9.0 | 5544 | 0.6613 | 0.8233 | 0.6984 | 0.6087 | 0.8190 |
67
+ | 0.5597 | 10.0 | 6160 | 0.6625 | 0.8256 | 0.6997 | 0.6138 | 0.8135 |
68
+
69
+
70
+ ### Framework versions
71
+
72
+ - Transformers 4.41.2
73
+ - Pytorch 2.3.0+cu121
74
+ - Datasets 2.19.2
75
+ - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4223000e3b3aed62b453b716aa084dcb45f4b9119eaa98fd6746fc1d544e04d8
3
  size 672708608
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7988e7b34ed830f32006a0f0de2c15aa18f1b0fa05bb78b6765053806ed0ee1
3
  size 672708608
runs/Jun12_13-59-23_1a9a5e08860f/events.out.tfevents.1718200764.1a9a5e08860f.1084.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:461ee1a901e83aac4deac4b6a28a53122b0937feaf4664eac15102edb7d43177
3
- size 12196
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc7e1f12d679eea5f510c91b3f4b9fe3ae3a2ced9c879db4c758b13a0469e917
3
+ size 12550