DunnBC22 commited on
Commit
debc549
1 Parent(s): 13e1aa3

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +62 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - bleu
7
+ - rouge
8
+ model-index:
9
+ - name: mbart-large-50-English_German_Translation
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # mbart-large-50-English_German_Translation
17
+
18
+ This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on the None dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 1.2342
21
+ - Bleu: 35.5931
22
+ - Rouge: {'rouge1': 0.5803386608353808, 'rouge2': 0.3939141514072567, 'rougeL': 0.5438629663406402, 'rougeLsum': 0.544153348468965}
23
+ - Meteor: {'meteor': 0.5500546034636025}
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 2e-05
43
+ - train_batch_size: 32
44
+ - eval_batch_size: 32
45
+ - seed: 42
46
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
+ - lr_scheduler_type: linear
48
+ - num_epochs: 1
49
+
50
+ ### Training results
51
+
52
+ | Training Loss | Epoch | Step | Validation Loss | Bleu | Rouge | Meteor |
53
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:--------------------------------------------------------------------------------------------------------------------------:|:------------------------------:|
54
+ | 1.7738 | 1.0 | 900 | 1.2342 | 35.7436 | {'rouge1': 0.5805815969432273, 'rouge2': 0.3941222478624937, 'rougeL': 0.544162316313326, 'rougeLsum': 0.5444260344836553} | {'meteor': 0.5511605039667078} |
55
+
56
+
57
+ ### Framework versions
58
+
59
+ - Transformers 4.22.2
60
+ - Pytorch 1.12.1
61
+ - Datasets 2.5.2
62
+ - Tokenizers 0.12.1