昇腾社区首页
中文
注册

generate_qa

功能描述

通过QAGenerationConfig传入标题和正文,会按照QAGenerationConfig中的max_tokens值截断正文。

大模型返回的QA如果不符合格式和个数要求,则会跳过。比如生成三个符合要求的QA:

Q1:如何查询成都火车站的停运列车?
参考段落:'查询方式:铁路12306网页首页。查询流程:第一步:进入铁路12306app首页,点击【车站大屏】;第二步:左上角车站名下拉选择成都东站;第三步:搜索框输入车次即可查询车次情况。'
Q2:四川省将洪水灾害防御响应提升至哪个级别?
参考段落:四川将洪水灾害防御四级响应提升至三级。
Q3:在7月14日,四川省气象台发布了哪种天气预警?
参考段落:7月14日15时30分,四川省气象台继续发布暴雨蓝色预警。

函数原型

def generate_qa(**kwargs)

参数说明

参数名

数据类型

可选/必选

说明

llm_config

LLMParameterConfig

可选

调用大模型的参数,此处修改默认值temperature为“0.5”,top_p为“0.95”,其余参数说明请参见LLMParameterConfig类

返回值说明

数据类型

说明

Dict

返回生成的QA对列表,格式为

{"从成都到重庆要多久? : 乘坐高铁1个小时",...}