Abdul Waheed
commited on
Commit
•
dd03f97
1
Parent(s):
ebfeb66
Update README.md
Browse files
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 |
+
---
|