← 返回列表

一种基于大语言模型的自然语句转SQL语句的方法

申请号: CN202410255498.X
申请人: 北京长河数智科技有限责任公司; 长河信息股份有限公司
申请日期: 2024/3/6

摘要文本

本申请公开了一种基于大语言模型的自然语句转SQL语句的方法,涉及自然语言技术领域,包括:从多个开放数据集中收集自然语句和对应的SQL语句,构建训练集;利用基于语法解析的模板和词汇替换规则,扩充训练集;构建序列到序列的神经网络,设置编码器对扩充后的训练集进行特征提取,设置解码器生成对应的SQL语句;在编码器中,设置表结构分析组件,从训练集中的SQL语句中解析对应的数据库表结构;在编码器中,设置数据库内容分析组件,从训练集中的SQL语句中解析对应的数据库表内容;在解码器中,设置数据库语法规则集,用于检查和修正生成的SQL语句中的语法错误。针对现有技术中自然语句到SQL语句的转换精度低的问题,本申请提升了模型的理解能力与转换精度。

专利详细信息

项目 内容
专利名称 一种基于大语言模型的自然语句转SQL语句的方法
专利类型 发明申请
申请号 CN202410255498.X
申请日 2024/3/6
公告号 CN117827886A
公开日 2024/4/5
IPC主分类号 G06F16/2452
权利人 北京长河数智科技有限责任公司; 长河信息股份有限公司
发明人 张煇; 剌昊跃; 李龙
地址 北京市丰台区汽车博物馆西路8号院3号楼7层705; 山西省太原市山西综改示范区太原学府园区东沺二巷3号高新国智大厦A座707室

专利主权项内容

1.一种基于大语言模型的自然语句转SQL语句的方法,包括:从多个开放数据集中收集自然语句和对应的SQL语句,构建训练集;在训练集上,利用基于语法解析的模板和词汇替换规则,扩充训练集;构建序列到序列的神经网络,设置编码器对扩充后的训练集进行特征提取,设置解码器生成对应的SQL语句;在编码器中,设置表结构分析组件,从训练集中的SQL语句中解析对应的数据库表结构,数据库表结构包含表名称、列名称和数据类型;在编码器中,设置数据库内容分析组件,从训练集中的SQL语句中解析对应的数据库表内容,数据库表内容包含数据值分布、是否包含空值和值区分度;在解码器中,设置数据库语法规则集,用于检查生成的SQL语句的语法错误并进行修正。