Mbart Large 50 English Spanish Translation

🚀 mbart-large-50-English_Spanish_Translation

このモデルは、facebook/mbart-large-50 をNoneデータセットでファインチューニングしたバージョンです。評価セットでは以下の結果を達成しています。

• 損失: 1.0290
• Bleu: 41.4437
• Rouge:
  • Rouge1: 0.6751402780531002
  • Rouge2: 0.49769602014143044
  • RougeL: 0.6371513427059108
  • RougeLsum: 0.6376403149816605
• Meteor: 0.6479226630466496

✨ 主な機能

このプロジェクトは、スペイン語のテキスト入力を英語に翻訳します。

このモデルを学習させるために作成したスクリプトのリンクはこちらです: https://github.com/DunnBC22/NLP_Projects/blob/main/Machine%20Translation/NLP%20Translation%20Project-EN:ES.ipynb

📚 ドキュメント

意図された用途と制限

このモデルは、技術を使って複雑な問題を解決する私の能力を示すことを目的としています。

学習と評価データ

データセットのソース: https://www.kaggle.com/datasets/hgultekin/paralel-translation-corpus-in-22-languages

学習手順

学習ハイパーパラメータ

学習中に以下のハイパーパラメータが使用されました。

• learning_rate: 2e-05
• train_batch_size: 32
• eval_batch_size: 32
• seed: 42
• optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
• lr_scheduler_type: linear
• num_epochs: 2

学習結果

フレームワークのバージョン

• Transformers 4.22.2
• Pytorch 1.12.1
• Datasets 2.5.2
• Tokenizers 0.12.1

📄 ライセンス

このモデルはMITライセンスの下で提供されています。