TheNewPing commited on
Commit
6c3795b
1 Parent(s): c17c302

End of training

Browse files
Files changed (1) hide show
  1. README.md +21 -1
README.md CHANGED
@@ -3,6 +3,11 @@ license: apache-2.0
3
  base_model: google/electra-base-discriminator
4
  tags:
5
  - generated_from_trainer
 
 
 
 
 
6
  model-index:
7
  - name: electra-base
8
  results: []
@@ -13,7 +18,13 @@ should probably proofread and complete it, then remove this comment. -->
13
 
14
  # electra-base
15
 
16
- This model is a fine-tuned version of [google/electra-base-discriminator](https://huggingface.co/google/electra-base-discriminator) on an unknown dataset.
 
 
 
 
 
 
17
 
18
  ## Model description
19
 
@@ -40,6 +51,15 @@ The following hyperparameters were used during training:
40
  - lr_scheduler_type: linear
41
  - num_epochs: 3
42
 
 
 
 
 
 
 
 
 
 
43
  ### Framework versions
44
 
45
  - Transformers 4.36.2
 
3
  base_model: google/electra-base-discriminator
4
  tags:
5
  - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ - precision
9
+ - recall
10
+ - f1
11
  model-index:
12
  - name: electra-base
13
  results: []
 
18
 
19
  # electra-base
20
 
21
+ This model is a fine-tuned version of [google/electra-base-discriminator](https://huggingface.co/google/electra-base-discriminator) on the None dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.3373
24
+ - Accuracy: 0.8875
25
+ - Precision: 0.8919
26
+ - Recall: 0.8818
27
+ - F1: 0.8868
28
 
29
  ## Model description
30
 
 
51
  - lr_scheduler_type: linear
52
  - num_epochs: 3
53
 
54
+ ### Training results
55
+
56
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
57
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
58
+ | 0.3187 | 1.0 | 1074 | 0.3036 | 0.8785 | 0.8880 | 0.8663 | 0.8770 |
59
+ | 0.228 | 2.0 | 2148 | 0.3022 | 0.8861 | 0.8921 | 0.8785 | 0.8853 |
60
+ | 0.1759 | 3.0 | 3222 | 0.3373 | 0.8875 | 0.8919 | 0.8818 | 0.8868 |
61
+
62
+
63
  ### Framework versions
64
 
65
  - Transformers 4.36.2