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个小时",...} |
父主题: 类QAGenerate