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

基于cointegrated/rubert-tiny2训练的rubert模型。

样本量: 4。

训练轮数: 16。

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