anidamarthy commited on
Commit
723752c
1 Parent(s): f38be55

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -5
README.md CHANGED
@@ -14,7 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Loss: 0.8568
18
 
19
  ## Model description
20
 
@@ -39,15 +39,22 @@ The following hyperparameters were used during training:
39
  - seed: 42
40
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
41
  - lr_scheduler_type: linear
42
- - num_epochs: 3.0
43
 
44
  ### Training results
45
 
46
  | Training Loss | Epoch | Step | Validation Loss |
47
  |:-------------:|:-----:|:----:|:---------------:|
48
- | 2.2498 | 1.0 | 3 | 1.2112 |
49
- | 1.3324 | 2.0 | 6 | 1.3771 |
50
- | 1.2313 | 3.0 | 9 | 0.8389 |
 
 
 
 
 
 
 
51
 
52
 
53
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Loss: 0.4294
18
 
19
  ## Model description
20
 
 
39
  - seed: 42
40
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
41
  - lr_scheduler_type: linear
42
+ - num_epochs: 10
43
 
44
  ### Training results
45
 
46
  | Training Loss | Epoch | Step | Validation Loss |
47
  |:-------------:|:-----:|:----:|:---------------:|
48
+ | 2.2498 | 1.0 | 3 | 1.1873 |
49
+ | 1.2651 | 2.0 | 6 | 1.2470 |
50
+ | 1.1038 | 3.0 | 9 | 0.6763 |
51
+ | 0.8357 | 4.0 | 12 | 0.5234 |
52
+ | 0.7296 | 5.0 | 15 | 0.4972 |
53
+ | 0.6506 | 6.0 | 18 | 0.4935 |
54
+ | 0.66 | 7.0 | 21 | 0.4576 |
55
+ | 0.5508 | 8.0 | 24 | 0.4905 |
56
+ | 0.5088 | 9.0 | 27 | 0.5120 |
57
+ | 0.4866 | 10.0 | 30 | 0.4520 |
58
 
59
 
60
  ### Framework versions