许可证: MIT

任务类型: 问答 示例: - 上下文: "普希金出生于1799年7月6日" - 问题: "普希金何时出生？" 示例标题: "测试"

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

样本量: 4。

训练轮次: 2。

from transformers import pipeline
qa_pipeline = pipeline(
    "question-answering",
    model="Den4ikAI/rubert-large-squad",
    tokenizer="Den4ikAI/rubert-large-squad"
)
predictions = qa_pipeline({
    'context': "普希金出生于1799年7月6日",
    'question': "普希金何时出生？"
})
print(predictions)
# 输出:
#{'score': 0.9613797664642334, 'start': 15, 'end': 31, 'answer': '1799年7月6日'}