Update README.md
Browse files
README.md
CHANGED
@@ -3,29 +3,46 @@ license: apache-2.0
|
|
3 |
base_model: Helsinki-NLP/opus-mt-de-en
|
4 |
tags:
|
5 |
- generated_from_trainer
|
|
|
6 |
model-index:
|
7 |
- name: opus-mt-de-en-OPUS_Medical_German_to_English
|
8 |
results: []
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
---
|
10 |
|
11 |
-
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
12 |
-
should probably proofread and complete it, then remove this comment. -->
|
13 |
-
|
14 |
# opus-mt-de-en-OPUS_Medical_German_to_English
|
15 |
|
16 |
-
This model is a fine-tuned version of [Helsinki-NLP/opus-mt-de-en](https://huggingface.co/Helsinki-NLP/opus-mt-de-en)
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
|
18 |
-
|
19 |
|
20 |
-
|
21 |
|
22 |
-
|
23 |
|
24 |
-
More information needed
|
25 |
|
26 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
27 |
|
28 |
-
More information needed
|
29 |
|
30 |
## Training procedure
|
31 |
|
@@ -44,11 +61,19 @@ The following hyperparameters were used during training:
|
|
44 |
|
45 |
### Training results
|
46 |
|
47 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
48 |
|
49 |
### Framework versions
|
50 |
|
51 |
- Transformers 4.31.0
|
52 |
- Pytorch 2.0.1+cu118
|
53 |
- Datasets 2.14.4
|
54 |
-
- Tokenizers 0.13.3
|
|
|
3 |
base_model: Helsinki-NLP/opus-mt-de-en
|
4 |
tags:
|
5 |
- generated_from_trainer
|
6 |
+
- medical
|
7 |
model-index:
|
8 |
- name: opus-mt-de-en-OPUS_Medical_German_to_English
|
9 |
results: []
|
10 |
+
datasets:
|
11 |
+
- ahazeemi/opus-medical-en-de
|
12 |
+
language:
|
13 |
+
- en
|
14 |
+
- de
|
15 |
+
metrics:
|
16 |
+
- bleu
|
17 |
+
- rouge
|
18 |
+
pipeline_tag: translation
|
19 |
---
|
20 |
|
|
|
|
|
|
|
21 |
# opus-mt-de-en-OPUS_Medical_German_to_English
|
22 |
|
23 |
+
This model is a fine-tuned version of [Helsinki-NLP/opus-mt-de-en](https://huggingface.co/Helsinki-NLP/opus-mt-de-en).
|
24 |
+
|
25 |
+
### Model description
|
26 |
+
|
27 |
+
For more information on how it was created, check out the following link: https://github.com/DunnBC22/NLP_Projects/blob/main/Machine%20Translation/Medical%20-%20German%20to%20English/OPUS_Medical_German_to_English_OPUS_Translation_Project.ipynb
|
28 |
+
|
29 |
+
### Intended uses & limitations
|
30 |
|
31 |
+
This model is intended to demonstrate my ability to solve a complex problem using technology.
|
32 |
|
33 |
+
### Training and evaluation data
|
34 |
|
35 |
+
Dataset Source: https://huggingface.co/datasets/ahazeemi/opus-medical-en-de
|
36 |
|
|
|
37 |
|
38 |
+
#### Histogram of German Input Word Counts
|
39 |
+
|
40 |
+
![German Word Count of Input Text](https://github.com/DunnBC22/NLP_Projects/raw/main/Machine%20Translation/Medical%20-%20German%20to%20English/Images/Histogram%20of%20German%20Input%20Lengths.png)
|
41 |
+
|
42 |
+
#### Histogram of English Input Word Counts
|
43 |
+
|
44 |
+
![English Word Count of Input Text](https://github.com/DunnBC22/NLP_Projects/raw/main/Machine%20Translation/Medical%20-%20German%20to%20English/Images/Histogram%20of%20English%20Input%20Lengths.png)
|
45 |
|
|
|
46 |
|
47 |
## Training procedure
|
48 |
|
|
|
61 |
|
62 |
### Training results
|
63 |
|
64 |
+
- eval_loss: 0.8723
|
65 |
+
- eval_bleu: 53.88120
|
66 |
+
- eval_rouge:
|
67 |
+
- rouge1: 0.7664
|
68 |
+
- rouge2: 0.6284
|
69 |
+
- rougeL: 0.7370
|
70 |
+
- rougeLsum: 0.7370
|
71 |
+
|
72 |
+
* The training results values are rounded to the nearest ten-thousandth.
|
73 |
|
74 |
### Framework versions
|
75 |
|
76 |
- Transformers 4.31.0
|
77 |
- Pytorch 2.0.1+cu118
|
78 |
- Datasets 2.14.4
|
79 |
+
- Tokenizers 0.13.3
|