ssilvera10 commited on
Commit
6d2fabb
·
verified ·
1 Parent(s): f12ea57

Model save

Browse files
Files changed (2) hide show
  1. README.md +7 -6
  2. model.safetensors +1 -1
README.md CHANGED
@@ -18,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.4704
22
- - Accuracy: 0.8137
23
- - F1: 0.8662
24
 
25
  ## Model description
26
 
@@ -39,19 +39,20 @@ More information needed
39
  ### Training hyperparameters
40
 
41
  The following hyperparameters were used during training:
42
- - learning_rate: 5e-05
43
  - train_batch_size: 8
44
  - eval_batch_size: 8
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
- - num_epochs: 2
49
 
50
  ### Training results
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
53
  |:-------------:|:------:|:----:|:---------------:|:--------:|:------:|
54
- | 0.6113 | 1.0893 | 500 | 0.4704 | 0.8137 | 0.8662 |
 
55
 
56
 
57
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.4265
22
+ - Accuracy: 0.8505
23
+ - F1: 0.8856
24
 
25
  ## Model description
26
 
 
39
  ### Training hyperparameters
40
 
41
  The following hyperparameters were used during training:
42
+ - learning_rate: 1e-05
43
  - train_batch_size: 8
44
  - eval_batch_size: 8
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
+ - num_epochs: 3
49
 
50
  ### Training results
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
53
  |:-------------:|:------:|:----:|:---------------:|:--------:|:------:|
54
+ | 0.5289 | 1.0893 | 500 | 0.3949 | 0.8529 | 0.8966 |
55
+ | 0.3665 | 2.1786 | 1000 | 0.4265 | 0.8505 | 0.8856 |
56
 
57
 
58
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7cf8ddd44f503b871a1839cf9afdba54be6dd6cba325c3a2d34ef49cf347e366
3
  size 498612824
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1379f2c8aed129157c3d409b3c6f33bf9cfccb00c986cccd50c43557e3f11762
3
  size 498612824