Update README.md
Browse files
README.md
CHANGED
@@ -42,33 +42,33 @@ widget:
|
|
42 |
|
43 |
This model is a fine-tuned version of [mdeberta-v3-base](https://huggingface.co/microsoft/mdeberta-v3-base) for a **novel extractive task**
|
44 |
which consists of **identifying the explanation of the correct answer** written by medical doctors. The model
|
45 |
-
has been fine-tuned using the multilingual [https://huggingface.co/datasets/HiTZ/casimedicos-squad](https://huggingface.co/datasets/HiTZ/casimedicos-squad) dataset
|
|
|
46 |
|
47 |
|
48 |
## Performance
|
49 |
|
50 |
-
F1 partial match
|
51 |
-
table:
|
52 |
|
53 |
-
|
54 |
|
55 |
-
###
|
56 |
|
57 |
The following hyperparameters were used during training:
|
58 |
- learning_rate: 5e-05
|
59 |
-
- train_batch_size:
|
60 |
- eval_batch_size: 8
|
61 |
-
- seed:
|
62 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
63 |
- lr_scheduler_type: linear
|
64 |
-
- num_epochs:
|
65 |
|
66 |
### Framework versions
|
67 |
|
68 |
-
- Transformers 4.
|
69 |
- Pytorch 2.1.2+cu121
|
70 |
- Datasets 2.16.1
|
71 |
- Tokenizers 0.15.2
|
72 |
|
73 |
-
**Contact**: [
|
74 |
HiTZ Center - Ixa, University of the Basque Country UPV/EHU
|
|
|
42 |
|
43 |
This model is a fine-tuned version of [mdeberta-v3-base](https://huggingface.co/microsoft/mdeberta-v3-base) for a **novel extractive task**
|
44 |
which consists of **identifying the explanation of the correct answer** written by medical doctors. The model
|
45 |
+
has been fine-tuned using the multilingual [https://huggingface.co/datasets/HiTZ/casimedicos-squad](https://huggingface.co/datasets/HiTZ/casimedicos-squad) dataset,
|
46 |
+
which includes English, French, Italian and Spanish.
|
47 |
|
48 |
|
49 |
## Performance
|
50 |
|
51 |
+
The model scores **74.64 F1 partial match** (as defined in [SQuAD extractive QA task](https://huggingface.co/datasets/rajpurkar/squad_v2) averaged across the 4 languages.
|
|
|
52 |
|
53 |
+
<!--<img src="https://raw.githubusercontent.com/hitz-zentroa/multilingual-abstrct/main/resources/multilingual-abstrct-results.png" style="width: 75%;"> -->
|
54 |
|
55 |
+
### Fine-tuning hyperparameters
|
56 |
|
57 |
The following hyperparameters were used during training:
|
58 |
- learning_rate: 5e-05
|
59 |
+
- train_batch_size: 48
|
60 |
- eval_batch_size: 8
|
61 |
+
- seed: random
|
62 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
63 |
- lr_scheduler_type: linear
|
64 |
+
- num_epochs: 20.0
|
65 |
|
66 |
### Framework versions
|
67 |
|
68 |
+
- Transformers 4.30.0.dev0
|
69 |
- Pytorch 2.1.2+cu121
|
70 |
- Datasets 2.16.1
|
71 |
- Tokenizers 0.15.2
|
72 |
|
73 |
+
**Contact**: [Iakes Goenaga](http://www.hitz.eus/es/node/65) and [Rodrigo Agerri](https://ragerri.github.io/)
|
74 |
HiTZ Center - Ixa, University of the Basque Country UPV/EHU
|