English To Urdu Translation Mbart

🚀 用於英語到烏爾都語翻譯的微調mBART模型

本倉庫包含一個經過微調的mBART模型，專門用於英語到烏爾都語的翻譯。該模型在自定義數據集上進行了訓練，並在測試數據上進行了評估。

🚀 快速開始

模型信息

使用示例

基礎用法

from transformers import MBart50TokenizerFast, MBartForConditionalGeneration

# 加載微調後的模型
model_name = "abdulwaheed1/english-to-urdu-translation-mbart"
tokenizer = MBart50TokenizerFast.from_pretrained(model_name, src_lang="en_XX", tgt_lang="ur_PK")
model = MBartForConditionalGeneration.from_pretrained(model_name)

評估

該模型在測試數據集上進行了評估，得到以下指標：

• BLEU分數：35.87
• 生成長度：42.56
• Meteor分數：0.60

訓練細節

該模型使用transformers庫進行訓練，配置如下：

• 訓練損失：1.5697
• 驗證損失：1.1256

數據集

該模型在自定義的英語 - 烏爾都語翻譯數據集上進行了微調。如果您希望使用相同的數據集，可以在data目錄中找到預處理腳本和數據集文件。

致謝

微調過程和代碼受到了 Hugging Face Transformers庫 的啟發。