Abdul Waheed commited on
Commit
dd03f97
1 Parent(s): ebfeb66

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +53 -1
README.md CHANGED
@@ -5,4 +5,56 @@ language:
5
  - ur
6
  metrics:
7
  - bleu
8
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  - ur
6
  metrics:
7
  - bleu
8
+ ---
9
+
10
+ ```markdown
11
+ # Fine-tuned mBART Model for English to Urdu Translation
12
+
13
+ This repository contains a fine-tuned mBART model for English to Urdu translation. The model has been trained on a custom dataset and evaluated on test data.
14
+
15
+ ## Model Information
16
+
17
+ - **Model Name:** `abdulwaheed63/mbart_en_ur_finetuned`
18
+ - **Base Model:** `facebook/mbart-large-50`
19
+ - **Tokenizer:** `facebook/mbart-large-50`
20
+ - **Source Language:** English (`en`)
21
+ - **Target Language:** Urdu (`ur`)
22
+
23
+ ## Usage
24
+
25
+ ```python
26
+ from transformers import MBart50TokenizerFast, MBartForConditionalGeneration
27
+
28
+ # Load the fine-tuned model
29
+ model_name = "abdulwaheed63/mbart_en_ur_finetuned"
30
+ tokenizer = MBart50TokenizerFast.from_pretrained(model_name, src_lang="en_XX", tgt_lang="ur_PK")
31
+ model = MBartForConditionalGeneration.from_pretrained(model_name)
32
+ ```
33
+
34
+ ## Evaluation
35
+
36
+ The model has been evaluated on a test dataset, and the following metrics were obtained:
37
+
38
+ - **BLEU Score:** 35.87
39
+ - **Generation Length:** 42.56
40
+ - **Meteor Score:** 0.60
41
+
42
+ ## Training Details
43
+
44
+ The model was trained using the `transformers` library with the following configuration:
45
+
46
+ - **Training Loss:** 1.5697
47
+ - **Validation Loss:** 1.1256
48
+
49
+
50
+ ## Dataset
51
+
52
+ The model was fine-tuned on a custom English-Urdu translation dataset. If you wish to use the same dataset, you can find the preprocessing script and dataset files in the `data` directory.
53
+
54
+ ## Acknowledgments
55
+
56
+ The fine-tuning process and code were inspired by the [Hugging Face Transformers library](https://github.com/huggingface/transformers).
57
+
58
+
59
+
60
+ ---