Rubert Large Squad

🚀 問答模型項目

本項目是一個問答模型，基於特定的訓練數據和模型架構，能夠高效準確地回答問題。

🚀 快速開始

本項目使用預訓練的 rubert 模型，由 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 года'}

📦 安裝指南

暫未提供安裝步驟相關內容。

💻 使用示例

基礎用法

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 года'}

🔧 技術細節

本模型基於 rubert 進行訓練，訓練數據規模為 4，訓練輪數為 2。

📄 許可證

本項目採用 MIT 許可證。