dminiotas05 commited on
Commit
9c319fd
1 Parent(s): 7d06f1e

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -10
README.md CHANGED
@@ -16,11 +16,11 @@ 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 an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 4.2038
20
- - Mse: 4.2038
21
- - Mae: 1.3920
22
- - R2: 0.4837
23
- - Accuracy: 0.2960
24
 
25
  ## Model description
26
 
@@ -45,14 +45,16 @@ The following hyperparameters were used during training:
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 | Mse | Mae | R2 | Accuracy |
53
- |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:--------:|
54
- | 4.7951 | 1.0 | 3122 | 4.3785 | 4.3785 | 1.4333 | 0.4622 | 0.2832 |
55
- | 3.4027 | 2.0 | 6244 | 4.2038 | 4.2038 | 1.3920 | 0.4837 | 0.2960 |
 
 
56
 
57
 
58
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 1.0940
20
+ - Mse: 4.3760
21
+ - Mae: 1.4084
22
+ - R2: 0.4625
23
+ - Accuracy: 0.3517
24
 
25
  ## Model description
26
 
 
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
+ - num_epochs: 4
49
 
50
  ### Training results
51
 
52
+ | Training Loss | Epoch | Step | Validation Loss | Mse | Mae | R2 | Accuracy |
53
+ |:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|:--------:|
54
+ | 0.7424 | 1.0 | 3122 | 1.1071 | 4.4286 | 1.4098 | 0.4561 | 0.3338 |
55
+ | 0.5038 | 2.0 | 6244 | 1.1794 | 4.7177 | 1.4140 | 0.4205 | 0.3677 |
56
+ | 0.356 | 3.0 | 9366 | 1.0717 | 4.2866 | 1.3852 | 0.4735 | 0.3581 |
57
+ | 0.2293 | 4.0 | 12488 | 1.0940 | 4.3760 | 1.4084 | 0.4625 | 0.3517 |
58
 
59
 
60
  ### Framework versions