ogimgio commited on
Commit
87c59f9
1 Parent(s): 031ad5a

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -6
README.md CHANGED
@@ -14,8 +14,8 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Loss: 3.0225
18
- - Mse: 3.5519
19
 
20
  ## Model description
21
 
@@ -40,15 +40,28 @@ The following hyperparameters were used during training:
40
  - seed: 42
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: constant
43
- - num_epochs: 3
44
 
45
  ### Training results
46
 
47
  | Training Loss | Epoch | Step | Validation Loss | Mse |
48
  |:-------------:|:-----:|:----:|:---------------:|:------:|
49
- | 11.289 | 1.0 | 3 | 6.5998 | 6.8064 |
50
- | 7.034 | 2.0 | 6 | 4.1748 | 4.1619 |
51
- | 3.708 | 3.0 | 9 | 3.0225 | 3.5519 |
 
 
 
 
 
 
 
 
 
 
 
 
 
52
 
53
 
54
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Loss: 0.2411
18
+ - Mse: 3.7748
19
 
20
  ## Model description
21
 
 
40
  - seed: 42
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: constant
43
+ - num_epochs: 30
44
 
45
  ### Training results
46
 
47
  | Training Loss | Epoch | Step | Validation Loss | Mse |
48
  |:-------------:|:-----:|:----:|:---------------:|:------:|
49
+ | 1.4559 | 1.0 | 122 | 0.6534 | 3.4024 |
50
+ | 0.5476 | 2.0 | 244 | 0.5601 | 3.8827 |
51
+ | 0.4224 | 3.0 | 366 | 0.4717 | 3.8263 |
52
+ | 0.3534 | 4.0 | 488 | 0.3511 | 3.7530 |
53
+ | 0.2827 | 5.0 | 610 | 0.2960 | 3.8889 |
54
+ | 0.2541 | 6.0 | 732 | 0.2416 | 3.5817 |
55
+ | 0.2289 | 7.0 | 854 | 0.3085 | 4.0660 |
56
+ | 0.1997 | 8.0 | 976 | 0.3212 | 3.4440 |
57
+ | 0.1889 | 9.0 | 1098 | 0.2852 | 3.9351 |
58
+ | 0.1752 | 10.0 | 1220 | 0.2360 | 3.8505 |
59
+ | 0.1683 | 11.0 | 1342 | 0.2939 | 4.1039 |
60
+ | 0.1601 | 12.0 | 1464 | 0.3242 | 4.0499 |
61
+ | 0.155 | 13.0 | 1586 | 0.2297 | 3.8442 |
62
+ | 0.1478 | 14.0 | 1708 | 0.2707 | 3.8680 |
63
+ | 0.1439 | 15.0 | 1830 | 0.2582 | 3.8703 |
64
+ | 0.1462 | 16.0 | 1952 | 0.2411 | 3.7748 |
65
 
66
 
67
  ### Framework versions