End of training
Browse files- README.md +22 -0
- model.safetensors +1 -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: trueparagraph.ai-ELECTRA
|
| 8 |
results: []
|
|
@@ -14,6 +19,16 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 14 |
# trueparagraph.ai-ELECTRA
|
| 15 |
|
| 16 |
This model is a fine-tuned version of [google/electra-base-discriminator](https://huggingface.co/google/electra-base-discriminator) on the None dataset.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
## Model description
|
| 19 |
|
|
@@ -43,6 +58,13 @@ The following hyperparameters were used during training:
|
|
| 43 |
|
| 44 |
### Training results
|
| 45 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
|
| 47 |
|
| 48 |
### Framework versions
|
|
|
|
| 3 |
base_model: google/electra-base-discriminator
|
| 4 |
tags:
|
| 5 |
- generated_from_trainer
|
| 6 |
+
metrics:
|
| 7 |
+
- accuracy
|
| 8 |
+
- f1
|
| 9 |
+
- precision
|
| 10 |
+
- recall
|
| 11 |
model-index:
|
| 12 |
- name: trueparagraph.ai-ELECTRA
|
| 13 |
results: []
|
|
|
|
| 19 |
# trueparagraph.ai-ELECTRA
|
| 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 |
+
- Accuracy: 0.9430
|
| 24 |
+
- F1: 0.9421
|
| 25 |
+
- Precision: 0.9528
|
| 26 |
+
- Recall: 0.9316
|
| 27 |
+
- Mcc: 0.8862
|
| 28 |
+
- Roc Auc: 0.9429
|
| 29 |
+
- Pr Auc: 0.9217
|
| 30 |
+
- Log Loss: 0.8825
|
| 31 |
+
- Loss: 0.2952
|
| 32 |
|
| 33 |
## Model description
|
| 34 |
|
|
|
|
| 58 |
|
| 59 |
### Training results
|
| 60 |
|
| 61 |
+
| Training Loss | Epoch | Step | Accuracy | F1 | Precision | Recall | Mcc | Roc Auc | Pr Auc | Log Loss | Validation Loss |
|
| 62 |
+
|:-------------:|:------:|:----:|:--------:|:------:|:---------:|:------:|:------:|:-------:|:------:|:--------:|:---------------:|
|
| 63 |
+
| 0.5401 | 0.6297 | 500 | 0.7694 | 0.7044 | 0.9732 | 0.5519 | 0.5963 | 0.7684 | 0.7602 | 3.5789 | 0.6109 |
|
| 64 |
+
| 0.3122 | 1.2594 | 1000 | 0.9225 | 0.9231 | 0.9122 | 0.9342 | 0.8452 | 0.9225 | 0.8850 | 1.1485 | 0.2368 |
|
| 65 |
+
| 0.2301 | 1.8892 | 1500 | 0.8670 | 0.8811 | 0.7942 | 0.9892 | 0.7573 | 0.8676 | 0.7910 | 1.9476 | 0.3654 |
|
| 66 |
+
| 0.1608 | 2.5189 | 2000 | 0.9348 | 0.9364 | 0.9103 | 0.9639 | 0.8711 | 0.9349 | 0.8955 | 1.0090 | 0.2677 |
|
| 67 |
+
| 0.1146 | 3.1486 | 2500 | 0.9430 | 0.9421 | 0.9528 | 0.9316 | 0.8862 | 0.9429 | 0.9217 | 0.8825 | 0.2952 |
|
| 68 |
|
| 69 |
|
| 70 |
### Framework versions
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 437959248
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:21b8970c12c7411141109c833e5158fe6bb31a13548a661944fbf368ffa3a0c3
|
| 3 |
size 437959248
|