simenaabol commited on
Commit
d83bd63
·
1 Parent(s): d30ab5e

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -10
README.md CHANGED
@@ -18,13 +18,13 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [bert-large-uncased](https://huggingface.co/bert-large-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.5844
22
  - Compute Metrics: :
23
- - Accuracy: 0.8
24
- - Balanced Accuracy: 0.5
25
- - F1 Score: 0.8889
26
- - Recall: 1.0
27
- - Precision: 0.8
28
 
29
  ## Model description
30
 
@@ -44,8 +44,8 @@ More information needed
44
 
45
  The following hyperparameters were used during training:
46
  - learning_rate: 5e-05
47
- - train_batch_size: 1
48
- - eval_batch_size: 1
49
  - seed: 42
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: linear
@@ -55,8 +55,8 @@ The following hyperparameters were used during training:
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Compute Metrics | Accuracy | Balanced Accuracy | F1 Score | Recall | Precision |
57
  |:-------------:|:-----:|:----:|:---------------:|:---------------:|:--------:|:-----------------:|:--------:|:------:|:---------:|
58
- | 0.5206 | 1.0 | 5 | 0.6186 | : | 0.8 | 0.5 | 0.8889 | 1.0 | 0.8 |
59
- | 0.5586 | 2.0 | 10 | 0.5844 | : | 0.8 | 0.5 | 0.8889 | 1.0 | 0.8 |
60
 
61
 
62
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [bert-large-uncased](https://huggingface.co/bert-large-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.6613
22
  - Compute Metrics: :
23
+ - Accuracy: 0.6
24
+ - Balanced Accuracy: 0.75
25
+ - F1 Score: 0.6667
26
+ - Recall: 0.5
27
+ - Precision: 1.0
28
 
29
  ## Model description
30
 
 
44
 
45
  The following hyperparameters were used during training:
46
  - learning_rate: 5e-05
47
+ - train_batch_size: 2
48
+ - eval_batch_size: 2
49
  - seed: 42
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: linear
 
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Compute Metrics | Accuracy | Balanced Accuracy | F1 Score | Recall | Precision |
57
  |:-------------:|:-----:|:----:|:---------------:|:---------------:|:--------:|:-----------------:|:--------:|:------:|:---------:|
58
+ | 1.0626 | 1.0 | 3 | 0.6567 | : | 0.8 | 0.5 | 0.8889 | 1.0 | 0.8 |
59
+ | 0.7417 | 2.0 | 6 | 0.6613 | : | 0.6 | 0.75 | 0.6667 | 0.5 | 1.0 |
60
 
61
 
62
  ### Framework versions