rendchevi commited on
Commit
eb8ed9a
·
verified ·
1 Parent(s): 399ce04

End of training

Browse files
Files changed (2) hide show
  1. README.md +18 -9
  2. model.safetensors +1 -1
README.md CHANGED
@@ -5,6 +5,8 @@ base_model: FacebookAI/roberta-base
5
  tags:
6
  - generated_from_trainer
7
  metrics:
 
 
8
  - accuracy
9
  model-index:
10
  - name: roberta-base-pr
@@ -18,9 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [FacebookAI/roberta-base](https://huggingface.co/FacebookAI/roberta-base) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.7464
22
- - Accuracy: 0.7575
23
- - F1 Macro: 0.5860
 
 
24
 
25
  ## Model description
26
 
@@ -45,15 +49,20 @@ The following hyperparameters were used during training:
45
  - seed: 42
46
  - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
47
  - lr_scheduler_type: linear
48
- - num_epochs: 3
49
 
50
  ### Training results
51
 
52
- | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Macro |
53
- |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|
54
- | No log | 1.0 | 240 | 0.9960 | 0.6608 | 0.4005 |
55
- | No log | 2.0 | 480 | 0.7770 | 0.7471 | 0.5742 |
56
- | 1.118 | 3.0 | 720 | 0.7464 | 0.7575 | 0.5860 |
 
 
 
 
 
57
 
58
 
59
  ### Framework versions
 
5
  tags:
6
  - generated_from_trainer
7
  metrics:
8
+ - precision
9
+ - recall
10
  - accuracy
11
  model-index:
12
  - name: roberta-base-pr
 
20
 
21
  This model is a fine-tuned version of [FacebookAI/roberta-base](https://huggingface.co/FacebookAI/roberta-base) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.8802
24
+ - F1 Macro: 0.6182
25
+ - Precision: 0.6222
26
+ - Recall: 0.6202
27
+ - Accuracy: 0.7742
28
 
29
  ## Model description
30
 
 
49
  - seed: 42
50
  - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
51
  - lr_scheduler_type: linear
52
+ - num_epochs: 8
53
 
54
  ### Training results
55
 
56
+ | Training Loss | Epoch | Step | Validation Loss | F1 Macro | Precision | Recall | Accuracy |
57
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:--------:|
58
+ | No log | 1.0 | 240 | 0.8923 | 0.4815 | 0.5010 | 0.5028 | 0.7107 |
59
+ | No log | 2.0 | 480 | 0.7876 | 0.5690 | 0.6087 | 0.5772 | 0.7477 |
60
+ | 1.0781 | 3.0 | 720 | 0.7258 | 0.6166 | 0.6277 | 0.6229 | 0.7758 |
61
+ | 1.0781 | 4.0 | 960 | 0.7538 | 0.6301 | 0.6368 | 0.6323 | 0.7804 |
62
+ | 0.5204 | 5.0 | 1200 | 0.7760 | 0.6278 | 0.6424 | 0.6297 | 0.7825 |
63
+ | 0.5204 | 6.0 | 1440 | 0.8245 | 0.6240 | 0.6264 | 0.6278 | 0.7789 |
64
+ | 0.3145 | 7.0 | 1680 | 0.8588 | 0.6156 | 0.6189 | 0.6184 | 0.7726 |
65
+ | 0.3145 | 8.0 | 1920 | 0.8802 | 0.6182 | 0.6222 | 0.6202 | 0.7742 |
66
 
67
 
68
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4be4cb21c3c9e89a325293eedcc1df0e21c9780ce4ed8271b1171af9e1b26f16
3
  size 498640508
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12759a1f8f394c2c439d011252a90bc40eb9e1387e5caea635bb5275b3645746
3
  size 498640508