phailyoor commited on
Commit
34ce176
1 Parent(s): d1a3720

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -8
README.md CHANGED
@@ -16,8 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 2.1342
20
- - Accuracy: 0.3707
21
 
22
  ## Model description
23
 
@@ -42,21 +42,22 @@ The following hyperparameters were used during training:
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
- - num_epochs: 4
46
 
47
  ### Training results
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
- | 2.2744 | 1.0 | 875 | 2.1780 | 0.3637 |
52
- | 2.0753 | 2.0 | 1750 | 2.1342 | 0.3707 |
53
- | 1.8624 | 3.0 | 2625 | 2.1430 | 0.3613 |
54
- | 1.7133 | 4.0 | 3500 | 2.1655 | 0.3593 |
 
55
 
56
 
57
  ### Framework versions
58
 
59
  - Transformers 4.12.3
60
- - Pytorch 1.9.0+cu111
61
  - Datasets 1.15.1
62
  - Tokenizers 0.10.3
 
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 2.4917
20
+ - Accuracy: 0.1472
21
 
22
  ## Model description
23
 
 
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
+ - num_epochs: 5
46
 
47
  ### Training results
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
+ | 2.7712 | 1.0 | 875 | 2.6037 | 0.1212 |
52
+ | 2.4896 | 2.0 | 1750 | 2.5123 | 0.1365 |
53
+ | 2.2153 | 3.0 | 2625 | 2.4918 | 0.142 |
54
+ | 1.93 | 4.0 | 3500 | 2.4917 | 0.1472 |
55
+ | 1.804 | 5.0 | 4375 | 2.5146 | 0.146 |
56
 
57
 
58
  ### Framework versions
59
 
60
  - Transformers 4.12.3
61
+ - Pytorch 1.9.0+cu102
62
  - Datasets 1.15.1
63
  - Tokenizers 0.10.3