数据集:

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

用于问答任务的Roberta大型模型(SQuAD 2.0版本)

本模型基于roberta-large架构开发。

训练数据

该模型使用SQuAD 2.0数据集进行训练。

适用于问答类任务场景。

使用方式与性能表现

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

from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline

# 加载模型及分词器
roberta_model = AutoModelForQuestionAnswering.from_pretrained('navteca/roberta-large-squad2')
roberta_tokenizer = AutoTokenizer.from_pretrained('navteca/roberta-large-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,
#}