许可证: MIT 管道标签: 问答 小部件: - 上下文: "普希金出生于1799年7月6日" - 文本: "普希金何时出生？" 示例标题: "测试"

基于sberbank-ai/ruBert-base训练的rubert模型。

样本量 - 4。

训练轮次 - 4。

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': "普希金出生于1799年7月6日",
    'question': "普希金何时出生？"
})
print(predictions)
# 输出:
#{'score': 0.8013797664642334, 'start': 15, 'end': 31, 'answer': '1799年7月6日'}