数据集:

• squad_v2 语言:
• en 库名称: transformers 管道标签: question-answering

基于Squad V2数据集微调的Mobile-Bert模型

该模型基于适用于便携设备或资源有限设备的mobile bert架构。

使用方法

首先使用transformers库加载模型和分词器:

from transformers import AutoModelForQuestionAnswering,  AutoTokenizer, pipeline

model_name = "aware-ai/mobilebert-squadv2"

model = AutoModelForQuestionAnswering.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

使用问答管道:

qa_engine = pipeline('question-answering', model=model, tokenizer=tokenizer)
QA_input = {
    'question': '你的问题?',
    'context': '你的上下文内容................'
}
res = qa_engine(QA_input)