数据集:

• squad_v2 语言: 英语 许可证: mit 任务标签: 问答 标签:
• roberta
• 问答

基于Roberta的问答模型(SQuAD 2.0版本)

本模型采用roberta-base架构。

训练数据

模型基于SQuAD 2.0数据集训练完成。

适用于问答类任务场景。

使用方式与性能表现

训练完成的模型可通过如下方式调用：

from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline

# 加载模型及分词器
roberta_model = AutoModelForQuestionAnswering.from_pretrained('navteca/roberta-base-squad2')
roberta_tokenizer = AutoTokenizer.from_pretrained('navteca/roberta-base-squad2')

# 获取预测结果
nlp = pipeline('question-answering', model=roberta_model, tokenizer=roberta_tokenizer)

result = nlp({
    'question': '柏林常住人口有多少？',
    'context': '柏林在891.82平方公里的区域内拥有3,520,031名注册居民。'
})

print(result)

#{
#  "answer": "3,520,031",
#  "end": 36,
#  "score": 0.96186668,
#  "start": 27,
#}