T5 Vi En Base

🚀 ベトナム語機械翻訳

本プロジェクトで提供されるモデルは、ベトナム語の機械翻訳に使用できます。先進的なアーキテクチャを利用して、効率的かつ正確な翻訳結果を実現し、ベトナム語関連の翻訳作業を強力にサポートします。

🚀 クイックスタート

モデルの説明

T5-vi-en-base は、T5 アーキテクチャを採用した、ベトナム語機械翻訳用の Transformer モデルです。

訓練データ

T5-vi-en-base は、400 万個の文ペア（英語 - ベトナム語）で訓練されています。

📦 インストール

ドキュメントに具体的なインストール手順が記載されていません。transformers ライブラリの公式インストールガイドを参考にしてインストールしてください。

💻 使用例

基本的な使用法

from transformers import T5ForConditionalGeneration, T5Tokenizer
import torch
if torch.cuda.is_available():       
    device = torch.device("cuda")

    print('There are %d GPU(s) available.' % torch.cuda.device_count())

    print('We will use the GPU:', torch.cuda.get_device_name(0))
else:
    print('No GPU available, using the CPU instead.')
    device = torch.device("cpu")

model = T5ForConditionalGeneration.from_pretrained("NlpHUST/t5-vi-en-base")
tokenizer = T5Tokenizer.from_pretrained("NlpHUST/t5-vi-en-base")
model.to(device)

src = "Theo lãnh đạo Sở Y tế, 3 người này không có triệu chứng sốt, ho, khó thở, đã được lấy mẫu xét nghiệm và cách ly tập trung."
tokenized_text = tokenizer.encode(src, return_tensors="pt").to(device)
model.eval()
summary_ids = model.generate(
                    tokenized_text,
                    max_length=256, 
                    num_beams=5,
                    repetition_penalty=2.5, 
                    length_penalty=1.0, 
                    early_stopping=True
                )
output = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
print(output)

According to the head of the Department of Health, the three people had no symptoms of fever, cough, shortness of breath, were taken samples for testing and concentrated quarantine.

出力結果の説明

上記のコードを実行すると、モデルは入力されたベトナム語の文を英語に翻訳して出力します。この例では、入力されたベトナム語の文 “Theo lãnh đạo Sở Y tế, 3 người này không có triệu chứng sốt, ho, khó thở, đã được lấy mẫu xét nghiệm và cách ly tập trung.” が “According to the head of the Department of Health, the three people had no symptoms of fever, cough, shortness of breath, were taken samples for testing and concentrated quarantine.” に翻訳されます。