Rubert Large Squad

🚀 質問応答モデル

このプロジェクトは、質問応答タスクに特化したモデルです。Sberbank-AIのruBert-baseをベースに学習され、高精度な応答を提供します。

🚀 クイックスタート

このモデルを使用するには、以下の手順に従ってください。

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

📦 インストール

必要なライブラリをインストールするには、以下のコマンドを実行します。

pip install transformers

💻 使用例

基本的な使用法

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

🔧 技術詳細

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

📄 ライセンス

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