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库 的启发。