Rubert Large Squad 2

🚀 質問応答モデル

このモデルは質問応答タスクに特化しており、Sberbank-aiのruBert-baseをベースに学習されています。

🚀 クイックスタート

このモデルは質問応答タスクに使用できます。以下のコードを使って簡単に推論を行うことができます。

💻 使用例

基本的な使用法

from transformers import pipeline
qa_pipeline = pipeline(
    "question-answering",
    model="Den4ikAI/rubert_large_squad_2",
    tokenizer="Den4ikAI/rubert_large_squad_2"
)
predictions = qa_pipeline({
    'context': "Пушкин родился 6 июля 1799 года",
    'question': "Когда родился Пушкин?"
})
print(predictions)
# output:
#{'score': 0.8013797664642334, 'start': 15, 'end': 31, 'answer': '6 июля 1799 года'}

🔧 技術詳細

• モデルはSberbank-aiのruBert-baseをベースに学習されています。
• 学習データのサンプルサイズは4です。
• 学習エポック数は4です。

📄 ライセンス

このプロジェクトはMITライセンスの下で公開されています。