dminiotas05 commited on
Commit
061c18c
1 Parent(s): be20b4d

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
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: 0.6410
20
- - Mse: 0.6410
21
- - Mae: 0.6143
22
- - R2: 0.3966
23
- - Accuracy: 0.5093
24
 
25
  ## Model description
26
 
@@ -40,8 +40,8 @@ More information needed
40
 
41
  The following hyperparameters were used during training:
42
  - learning_rate: 2e-05
43
- - train_batch_size: 16
44
- - eval_batch_size: 16
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
@@ -51,7 +51,7 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Mse | Mae | R2 | Accuracy |
53
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:--------:|
54
- | 1.0465 | 1.0 | 750 | 0.6410 | 0.6410 | 0.6143 | 0.3966 | 0.5093 |
55
 
56
 
57
  ### 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: 0.6143
20
+ - Mse: 0.6143
21
+ - Mae: 0.6022
22
+ - R2: 0.4218
23
+ - Accuracy: 0.52
24
 
25
  ## Model description
26
 
 
40
 
41
  The following hyperparameters were used during training:
42
  - learning_rate: 2e-05
43
+ - train_batch_size: 64
44
+ - eval_batch_size: 64
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Mse | Mae | R2 | Accuracy |
53
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:--------:|
54
+ | 0.5241 | 1.0 | 188 | 0.6143 | 0.6143 | 0.6022 | 0.4218 | 0.52 |
55
 
56
 
57
  ### Framework versions