Update README.md
Browse files
README.md
CHANGED
@@ -11,8 +11,11 @@ probably proofread and complete it, then remove this comment. -->
|
|
11 |
|
12 |
# electra-nli_finetuned
|
13 |
|
14 |
-
|
15 |
-
It
|
|
|
|
|
|
|
16 |
|
17 |
|
18 |
## Model description
|
@@ -31,12 +34,17 @@ More information needed
|
|
31 |
|
32 |
### Training hyperparameters
|
33 |
|
34 |
-
The following hyperparameters were used during training:
|
35 |
-
- optimizer: None
|
36 |
-
- training_precision: float32
|
37 |
|
38 |
### Training results
|
39 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
40 |
|
41 |
|
42 |
### Framework versions
|
|
|
11 |
|
12 |
# electra-nli_finetuned
|
13 |
|
14 |
+
The model base is [electra-small](google/electra-small-discriminator).
|
15 |
+
It has been fine-tuned on: [snli](https://huggingface.co/datasets/snli), [scitail](https://huggingface.co/datasets/scitail),
|
16 |
+
[wanli](https://huggingface.co/datasets/alisawuffles/WANLI),
|
17 |
+
[mnli](https://huggingface.co/datasets/multi_nli), [anli](https://huggingface.co/datasets/anli).
|
18 |
+
|
19 |
|
20 |
|
21 |
## Model description
|
|
|
34 |
|
35 |
### Training hyperparameters
|
36 |
|
|
|
|
|
|
|
37 |
|
38 |
### Training results
|
39 |
+
It achieved the following accuracy during training:
|
40 |
+
snli: 89.15
|
41 |
+
scitail: 90.08%
|
42 |
+
wanli: 67.84%
|
43 |
+
mnli: 81.95%
|
44 |
+
fever-nli: 74.14%
|
45 |
+
anli-r1_test: 46.60%
|
46 |
+
anli-r2_test: 42.50%
|
47 |
+
anli-r3_test: 43.08%
|
48 |
|
49 |
|
50 |
### Framework versions
|