-
公开(公告)号:CN117609279A
公开(公告)日:2024-02-27
申请号:CN202311788083.0
申请日:2023-12-22
申请人: 杭州西湖数据智能研究院
IPC分类号: G06F16/242 , G06F16/22 , G06F16/28
摘要: 本发明提供了一种基于LLM模型的NL2SQL方法及存储介质,涉及自然语言处理领域,所述方法包括:获取训练用自然语句列表和训练用SQL语句列表,获取预测SQL语句列表并对预测SQL语句进行拆分,获取预测SQL语句对应的键值对,获取预测SQL语句对应的难度等级,获取任意难度等级的预测SQL语句列表且计算该难度等级的预测SQL语句列表对应的损失,若对任意难度等级的预测SQL语句列表的损失均小于对应的损失阈值,将初始LLM模型作为目标LLM模型,使用目标LLM模型获取目标语句对应的SQL语句;本发明通过对预测SQL语句划分为不同的难度等级,达到对LLM模型转化SQL语句的特定难度等级的训练。