0xMaka commited on
Commit
3724c05
1 Parent(s): e858bdf

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -7
README.md CHANGED
@@ -21,10 +21,10 @@ model-index:
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
- value: 0.8233333333333334
25
  - name: F1
26
  type: f1
27
- value: 0.8306709265175719
28
  ---
29
 
30
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -34,9 +34,9 @@ should probably proofread and complete it, then remove this comment. -->
34
 
35
  This model was trained from scratch on the imdb dataset.
36
  It achieves the following results on the evaluation set:
37
- - Loss: 0.4381
38
- - Accuracy: 0.8233
39
- - F1: 0.8307
40
 
41
  ## Model description
42
 
@@ -56,8 +56,8 @@ More information needed
56
 
57
  The following hyperparameters were used during training:
58
  - learning_rate: 2e-05
59
- - train_batch_size: 4
60
- - eval_batch_size: 4
61
  - seed: 42
62
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
63
  - lr_scheduler_type: linear
 
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
+ value: 0.8333333333333334
25
  - name: F1
26
  type: f1
27
+ value: 0.8355263157894737
28
  ---
29
 
30
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
34
 
35
  This model was trained from scratch on the imdb dataset.
36
  It achieves the following results on the evaluation set:
37
+ - Loss: 0.4277
38
+ - Accuracy: 0.8333
39
+ - F1: 0.8355
40
 
41
  ## Model description
42
 
 
56
 
57
  The following hyperparameters were used during training:
58
  - learning_rate: 2e-05
59
+ - train_batch_size: 8
60
+ - eval_batch_size: 8
61
  - seed: 42
62
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
63
  - lr_scheduler_type: linear