一种面向函数调用关系的代码预训练及生成的方法与系统

本发明公开了一种面向函数调用关系的代码预训练及生成的方法与系统。本方法为：1)对于代码数据集中的每一条代码仓库数据，从该条代码仓库数据中提取上下文信息；所述上下文信息为函数调用信息、参数依赖及全局逻辑结构；2)利用所述上下文信息训练代码相关的深度学习模型，得到代码生成模型；3)将代码需求描述或待编辑代码输入所述代码生成模型，生成对应的目标代码；其中，对于输入的代码需求描述，所述代码生成模型根据代码需求描述生成符合需求的目标代码；对于输入的待编辑代码，所述代码生成模型对待编辑代码进行自动修复或优化得到目标代码。本发明优化了生成代码的逻辑一致性和可用性，显著提升了复杂代码场景下的代码生成。