一种基于增强提示学习框架的源代码摘要生成方法及系统

本发明公开一种基于增强提示学习框架的源代码摘要生成方法及系统，将代码片段输入至增强提示学习框架进行处理，生成自然语言形式的代码摘要；增强提示学习框架包括代码预训练模型、映射模块、结构代理模块和大语言模型；所述代码预训练模型接收代码片段并进行特征提取，得到代码特征，将所述代码特征输入至映射模块，生成知识提示向量；同时将所述代码特征输入至结构代理模块，生成结构提示向量；将所述知识提示向量与所述结构提示向量拼接后输入至大语言模型作为提示信息，生成自然语言形式的代码摘要。通过结合代码预训练模型的结构理解能力与大语言模型的自然语言生成能力，生成更为精准的代码摘要。