Opus Mt De En OPUS Medical German To English

🚀 医学德语到英语翻译模型 opus-mt-de-en-OPUS_Medical_German_to_English

本模型是一个可将医学德语翻译成英语的模型，基于 Helsinki-NLP/opus-mt-de-en 微调而来，能很好地展示运用技术解决复杂问题的能力。

🚀 快速开始

此模型是 Helsinki-NLP/opus-mt-de-en 的微调版本。如需了解其创建方式的更多信息，请查看以下链接：https://github.com/DunnBC22/NLP_Projects/blob/main/Machine%20Translation/Medical%20-%20German%20to%20English/OPUS_Medical_German_to_English_OPUS_Translation_Project.ipynb

✨ 主要特性

• 特定领域翻译：专注于医学领域的德语到英语翻译。
• 微调优化：基于预训练模型微调，更贴合医学翻译需求。

📚 详细文档

预期用途与限制

本模型旨在展示使用技术解决复杂问题的能力。

训练和评估数据

• 数据集来源：https://huggingface.co/datasets/ahazeemi/opus-medical-en-de

德语输入词数直方图

英语输入词数直方图

🔧 技术细节

训练过程

训练超参数

训练期间使用了以下超参数：

• 学习率（learning_rate）：2e - 05
• 训练批次大小（train_batch_size）：8
• 评估批次大小（eval_batch_size）：8
• 随机种子（seed）：42
• 梯度累积步数（gradient_accumulation_steps）：4
• 总训练批次大小（total_train_batch_size）：32
• 优化器（optimizer）：Adam，β=(0.9, 0.999)，ε = 1e - 08
• 学习率调度器类型（lr_scheduler_type）：线性
• 训练轮数（num_epochs）：5

训练结果

• 评估损失（eval_loss）：0.8723
• 评估 BLEU 分数（eval_bleu）：53.88120
• 评估 ROUGE 分数：
  • ROUGE - 1（rouge1）：0.7664
  • ROUGE - 2（rouge2）：0.6284
  • ROUGE - L（rougeL）：0.7370
  • ROUGE - Lsum（rougeLsum）：0.7370

• 训练结果值四舍五入到小数点后四位。

框架版本

• Transformers 4.31.0
• Pytorch 2.0.1+cu118
• Datasets 2.14.4
• Tokenizers 0.13.3

📄 许可证

本模型采用 Apache - 2.0 许可证。

📋 模型信息表格