ytonthat commited on
Commit
4e0462c
1 Parent(s): a3f8b16

Training complete

Browse files
README.md CHANGED
@@ -25,16 +25,16 @@ model-index:
25
  metrics:
26
  - name: Precision
27
  type: precision
28
- value: 0.5151515151515151
29
  - name: Recall
30
  type: recall
31
- value: 0.5749888740542947
32
  - name: F1
33
  type: f1
34
- value: 0.5434279705573082
35
  - name: Accuracy
36
  type: accuracy
37
- value: 0.8052467462039046
38
  ---
39
 
40
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -44,11 +44,11 @@ should probably proofread and complete it, then remove this comment. -->
44
 
45
  This model is a fine-tuned version of [Dr-BERT/DrBERT-7GB](https://huggingface.co/Dr-BERT/DrBERT-7GB) on the quaero dataset.
46
  It achieves the following results on the evaluation set:
47
- - Loss: 1.0979
48
- - Precision: 0.5152
49
- - Recall: 0.5750
50
- - F1: 0.5434
51
- - Accuracy: 0.8052
52
 
53
  ## Model description
54
 
@@ -67,26 +67,28 @@ More information needed
67
  ### Training hyperparameters
68
 
69
  The following hyperparameters were used during training:
70
- - learning_rate: 5e-05
71
  - train_batch_size: 8
72
  - eval_batch_size: 8
73
  - seed: 42
74
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
75
  - lr_scheduler_type: linear
76
- - num_epochs: 8
77
 
78
  ### Training results
79
 
80
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
81
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
82
- | No log | 1.0 | 105 | 0.7366 | 0.4295 | 0.4815 | 0.4540 | 0.7706 |
83
- | No log | 2.0 | 210 | 0.7073 | 0.4751 | 0.5256 | 0.4990 | 0.7844 |
84
- | No log | 3.0 | 315 | 0.7970 | 0.5277 | 0.5087 | 0.5180 | 0.7960 |
85
- | No log | 4.0 | 420 | 0.9072 | 0.4960 | 0.5287 | 0.5118 | 0.7926 |
86
- | 0.4094 | 5.0 | 525 | 0.9859 | 0.4765 | 0.5696 | 0.5190 | 0.7899 |
87
- | 0.4094 | 6.0 | 630 | 1.0399 | 0.5093 | 0.5750 | 0.5401 | 0.8039 |
88
- | 0.4094 | 7.0 | 735 | 1.0778 | 0.5245 | 0.5723 | 0.5474 | 0.8065 |
89
- | 0.4094 | 8.0 | 840 | 1.0979 | 0.5152 | 0.5750 | 0.5434 | 0.8052 |
 
 
90
 
91
 
92
  ### Framework versions
 
25
  metrics:
26
  - name: Precision
27
  type: precision
28
+ value: 0.502933124755573
29
  - name: Recall
30
  type: recall
31
+ value: 0.5723186470850022
32
  - name: F1
33
  type: f1
34
+ value: 0.5353871773522064
35
  - name: Accuracy
36
  type: accuracy
37
+ value: 0.7962988069414316
38
  ---
39
 
40
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
44
 
45
  This model is a fine-tuned version of [Dr-BERT/DrBERT-7GB](https://huggingface.co/Dr-BERT/DrBERT-7GB) on the quaero dataset.
46
  It achieves the following results on the evaluation set:
47
+ - Loss: 1.2469
48
+ - Precision: 0.5029
49
+ - Recall: 0.5723
50
+ - F1: 0.5354
51
+ - Accuracy: 0.7963
52
 
53
  ## Model description
54
 
 
67
  ### Training hyperparameters
68
 
69
  The following hyperparameters were used during training:
70
+ - learning_rate: 0.0001
71
  - train_batch_size: 8
72
  - eval_batch_size: 8
73
  - seed: 42
74
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
75
  - lr_scheduler_type: linear
76
+ - num_epochs: 10
77
 
78
  ### Training results
79
 
80
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
81
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
82
+ | No log | 1.0 | 105 | 0.7546 | 0.4019 | 0.4766 | 0.4361 | 0.7598 |
83
+ | No log | 2.0 | 210 | 0.7426 | 0.4483 | 0.4980 | 0.4719 | 0.7785 |
84
+ | No log | 3.0 | 315 | 0.8678 | 0.4413 | 0.5349 | 0.4836 | 0.7694 |
85
+ | No log | 4.0 | 420 | 0.9466 | 0.4484 | 0.5336 | 0.4873 | 0.7780 |
86
+ | 0.4639 | 5.0 | 525 | 1.0138 | 0.4767 | 0.5194 | 0.4971 | 0.7864 |
87
+ | 0.4639 | 6.0 | 630 | 1.0300 | 0.4831 | 0.5269 | 0.5040 | 0.7820 |
88
+ | 0.4639 | 7.0 | 735 | 1.1711 | 0.4841 | 0.5572 | 0.5181 | 0.7855 |
89
+ | 0.4639 | 8.0 | 840 | 1.1943 | 0.5117 | 0.5434 | 0.5271 | 0.7960 |
90
+ | 0.4639 | 9.0 | 945 | 1.2376 | 0.5115 | 0.5719 | 0.5400 | 0.7955 |
91
+ | 0.0379 | 10.0 | 1050 | 1.2469 | 0.5029 | 0.5723 | 0.5354 | 0.7963 |
92
 
93
 
94
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0be7caba0fd6fb8b10f8211211f7b51b2298bfa0ba982cde2f12398244fccddb
3
  size 440213956
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:688464a45097edd10be0f8d65b6935d7660d877b3026d414b0515b818b716189
3
  size 440213956
runs/Mar01_14-35-57_6d7974edcc8d/events.out.tfevents.1709303758.6d7974edcc8d.7770.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a15d33fcdcf0d45c27825f938c6133a0ae39af45ea77d07b8c8b2a764f65958b
3
- size 9419
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e8ca518dcd0a1aa486c7a70b3a4c56a9c1f49cbdd95738b210255d033147444
3
+ size 10928