Pablo94 commited on
Commit
84350a9
1 Parent(s): cd4c3db

End of training

Browse files
Files changed (2) hide show
  1. README.md +75 -0
  2. pytorch_model.bin +1 -1
README.md ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: google/electra-base-discriminator
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ - precision
9
+ - recall
10
+ model-index:
11
+ - name: electra-base-discriminator-finetuned-detests
12
+ results: []
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+ # electra-base-discriminator-finetuned-detests
19
+
20
+ This model is a fine-tuned version of [google/electra-base-discriminator](https://huggingface.co/google/electra-base-discriminator) on the None dataset.
21
+ It achieves the following results on the evaluation set:
22
+ - Loss: 1.1215
23
+ - Accuracy: 0.7807
24
+ - F1-score: 0.7308
25
+ - Precision: 0.7162
26
+ - Recall: 0.7768
27
+ - Auc: 0.7768
28
+
29
+ ## Model description
30
+
31
+ More information needed
32
+
33
+ ## Intended uses & limitations
34
+
35
+ More information needed
36
+
37
+ ## Training and evaluation data
38
+
39
+ More information needed
40
+
41
+ ## Training procedure
42
+
43
+ ### Training hyperparameters
44
+
45
+ The following hyperparameters were used during training:
46
+ - learning_rate: 2e-05
47
+ - train_batch_size: 16
48
+ - eval_batch_size: 16
49
+ - seed: 42
50
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
+ - lr_scheduler_type: linear
52
+ - num_epochs: 10
53
+
54
+ ### Training results
55
+
56
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1-score | Precision | Recall | Auc |
57
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|:---------:|:------:|:------:|
58
+ | 0.3236 | 1.0 | 174 | 0.4661 | 0.7610 | 0.6684 | 0.6647 | 0.6728 | 0.6728 |
59
+ | 0.3239 | 2.0 | 348 | 0.4287 | 0.7987 | 0.7144 | 0.7138 | 0.7149 | 0.7149 |
60
+ | 0.3421 | 3.0 | 522 | 0.5586 | 0.7741 | 0.7292 | 0.7163 | 0.7853 | 0.7853 |
61
+ | 0.2288 | 4.0 | 696 | 0.6229 | 0.7807 | 0.7308 | 0.7162 | 0.7768 | 0.7768 |
62
+ | 0.1888 | 5.0 | 870 | 0.6629 | 0.7954 | 0.7293 | 0.7173 | 0.7483 | 0.7483 |
63
+ | 0.2205 | 6.0 | 1044 | 0.8462 | 0.8036 | 0.7349 | 0.7251 | 0.7485 | 0.7485 |
64
+ | 0.1512 | 7.0 | 1218 | 0.8362 | 0.8151 | 0.7335 | 0.7367 | 0.7306 | 0.7306 |
65
+ | 0.2345 | 8.0 | 1392 | 1.0372 | 0.7758 | 0.7204 | 0.7063 | 0.7584 | 0.7584 |
66
+ | 0.0592 | 9.0 | 1566 | 1.0396 | 0.7840 | 0.7291 | 0.7142 | 0.7663 | 0.7663 |
67
+ | 0.0381 | 10.0 | 1740 | 1.1215 | 0.7807 | 0.7308 | 0.7162 | 0.7768 | 0.7768 |
68
+
69
+
70
+ ### Framework versions
71
+
72
+ - Transformers 4.33.1
73
+ - Pytorch 2.0.1+cu118
74
+ - Datasets 2.14.5
75
+ - Tokenizers 0.13.3
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8be4a1871d69a27457e4d47f86f1c59f321c5e81c12874d28af07c15369f0d2a
3
  size 438004081
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7440745613bf3856af72d4a26cf49e8e008446f2eed67096e8a60e69cdd0807
3
  size 438004081