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個epoch。

📄 許可證

本項目採用MIT許可證。