一种三阶段训练NL2SQL模型的方法

本申请公开了一种三阶段训练NL2SQL模型的方法，其包括：对原始数据进行预处理得到训练数据集，初始化模型；所述原始数据包括数据库表名、创建数据库的sql语句、用户问题、标准SQL语句、检索内容；冷启动阶段，加载Base模型、Lora_init模型和训练数据集，训练完成，得到Lora_Ice模型；Lora_Ice模型作为GRPO阶段的初始Lora模型；GRPO阶段，加载Base模型、Lora_Ice模型和训练数据集，训练完成，得到Lora_GRPO模型，在训练过程中收集反馈数据；反馈阶段，加载Base模型、Lora_init模型和反馈数据集，训练完成，得到Lora_Feedback模型。本申请提升了自然语言到SQL语句的转换准确性与执行效率，能够根据用户输入的问题，快速输出正确的答案。