gokuls commited on
Commit
cc662f0
1 Parent(s): 0212f34

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -18
README.md CHANGED
@@ -1,6 +1,4 @@
1
  ---
2
- language:
3
- - en
4
  tags:
5
  - generated_from_trainer
6
  datasets:
@@ -15,7 +13,7 @@ model-index:
15
  name: Text Classification
16
  type: text-classification
17
  dataset:
18
- name: GLUE MRPC
19
  type: glue
20
  config: mrpc
21
  split: validation
@@ -23,10 +21,10 @@ model-index:
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
- value: 0.6838235294117647
27
  - name: F1
28
  type: f1
29
- value: 0.8122270742358079
30
  ---
31
 
32
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -34,12 +32,12 @@ should probably proofread and complete it, then remove this comment. -->
34
 
35
  # hBERTv2_new_pretrain_mrpc
36
 
37
- This model is a fine-tuned version of [gokuls/bert_12_layer_model_v2_complete_training_new](https://huggingface.co/gokuls/bert_12_layer_model_v2_complete_training_new) on the GLUE MRPC dataset.
38
  It achieves the following results on the evaluation set:
39
- - Loss: 1.8707
40
- - Accuracy: 0.6838
41
- - F1: 0.8122
42
- - Combined Score: 0.7480
43
 
44
  ## Model description
45
 
@@ -58,7 +56,7 @@ More information needed
58
  ### Training hyperparameters
59
 
60
  The following hyperparameters were used during training:
61
- - learning_rate: 0.0005
62
  - train_batch_size: 128
63
  - eval_batch_size: 128
64
  - seed: 10
@@ -71,13 +69,14 @@ The following hyperparameters were used during training:
71
 
72
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Combined Score |
73
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:--------------:|
74
- | 8.6576 | 1.0 | 29 | 3.5192 | 0.6838 | 0.8122 | 0.7480 |
75
- | 9.1979 | 2.0 | 58 | 1.8707 | 0.6838 | 0.8122 | 0.7480 |
76
- | 8.8322 | 3.0 | 87 | 3.9709 | 0.6838 | 0.8122 | 0.7480 |
77
- | 8.8403 | 4.0 | 116 | 3.3888 | 0.6838 | 0.8122 | 0.7480 |
78
- | 8.3428 | 5.0 | 145 | 2.3995 | 0.6838 | 0.8122 | 0.7480 |
79
- | 8.565 | 6.0 | 174 | 5.375 | 0.6838 | 0.8122 | 0.7480 |
80
- | 12.6976 | 7.0 | 203 | 14.0184 | 0.3162 | 0.0 | 0.1581 |
 
81
 
82
 
83
  ### Framework versions
 
1
  ---
 
 
2
  tags:
3
  - generated_from_trainer
4
  datasets:
 
13
  name: Text Classification
14
  type: text-classification
15
  dataset:
16
+ name: glue
17
  type: glue
18
  config: mrpc
19
  split: validation
 
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
+ value: 0.625
25
  - name: F1
26
  type: f1
27
+ value: 0.7118644067796611
28
  ---
29
 
30
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
32
 
33
  # hBERTv2_new_pretrain_mrpc
34
 
35
+ This model is a fine-tuned version of [gokuls/bert_12_layer_model_v2_complete_training_new](https://huggingface.co/gokuls/bert_12_layer_model_v2_complete_training_new) on the glue dataset.
36
  It achieves the following results on the evaluation set:
37
+ - Loss: 1.3516
38
+ - Accuracy: 0.625
39
+ - F1: 0.7119
40
+ - Combined Score: 0.6684
41
 
42
  ## Model description
43
 
 
56
  ### Training hyperparameters
57
 
58
  The following hyperparameters were used during training:
59
+ - learning_rate: 4e-05
60
  - train_batch_size: 128
61
  - eval_batch_size: 128
62
  - seed: 10
 
69
 
70
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Combined Score |
71
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:--------------:|
72
+ | 0.6721 | 1.0 | 29 | 0.6200 | 0.6838 | 0.8122 | 0.7480 |
73
+ | 0.6229 | 2.0 | 58 | 0.6098 | 0.6569 | 0.7255 | 0.6912 |
74
+ | 0.5689 | 3.0 | 87 | 0.5990 | 0.7034 | 0.8118 | 0.7576 |
75
+ | 0.4615 | 4.0 | 116 | 0.6689 | 0.6765 | 0.78 | 0.7282 |
76
+ | 0.3475 | 5.0 | 145 | 0.8472 | 0.6054 | 0.6774 | 0.6414 |
77
+ | 0.2307 | 6.0 | 174 | 0.9917 | 0.6103 | 0.6913 | 0.6508 |
78
+ | 0.166 | 7.0 | 203 | 1.1149 | 0.6544 | 0.7522 | 0.7033 |
79
+ | 0.1258 | 8.0 | 232 | 1.3516 | 0.625 | 0.7119 | 0.6684 |
80
 
81
 
82
  ### Framework versions