English To Urdu Translation Mbart

🚀 英語からウルドゥー語への翻訳用のFine-tuned mBARTモデル

このリポジトリには、英語からウルドゥー語への翻訳用にFine-tuningされたmBARTモデルが含まれています。このモデルはカスタムデータセットで学習され、テストデータで評価されています。

🚀 クイックスタート

このFine-tuned mBARTモデルを使用することで、英語からウルドゥー語への高精度な翻訳が可能です。以下のセクションでは、モデルの情報、使用方法、評価結果などを詳しく説明します。

✨ 主な機能

• 英語からウルドゥー語への翻訳機能を提供します。
• カスタムデータセットで学習されたFine-tunedモデルで、高精度な翻訳が期待できます。
• transformersライブラリを使用しているため、使いやすく柔軟性が高いです。

📦 インストール

このモデルを使用するには、transformersライブラリが必要です。以下のコマンドでインストールできます。

pip install transformers

💻 使用例

基本的な使用法

from transformers import MBart50TokenizerFast, MBartForConditionalGeneration

# Load the fine-tuned model
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

データセット

このモデルはカスタムの英語 - ウルドゥー語翻訳データセットでFine-tuningされています。同じデータセットを使用したい場合は、dataディレクトリに前処理スクリプトとデータセットファイルがあります。

謝辞

Fine-tuningのプロセスとコードは、Hugging Face Transformersライブラリにインスパイアされています。