一种基于对偶图神经网络的代码注释生成方法和系统

    公开(公告)号:CN119473383A

    公开(公告)日:2025-02-18

    申请号:CN202411583315.3

    申请日:2024-11-07

    Abstract: 本发明涉及软件工程技术领域,公开了一种基于对偶图神经网络的代码注释生成方法和系统,方法包括:获取源代码以及作为源代码的自然语言注释,并对源代码进行预处理,根据预处理之后的源代码和自然语言注释构建训练数据;构建代码处理模型,使用训练数据训练代码处理模型,获得训练好的代码处理模型,代码处理模型包括对偶图模块、编码器、聚合器和解码器,对偶图模块根据预处理之后的源代码解析获取抽象语法树,并根据抽象语法树构建语法依赖图和语义依赖图;编码器根据所述语法依赖图获得第一编码表示、根据语义依赖图获得第二编码表示;聚合器根据第一编码表示和第二编码表示获得聚合图表示;解码器根据第一编码表示、第二编码表示和聚合图表示生成自然语言注释;将待注释的代码输入训练好的代码处理模型,获得自然语言注释。本发明能够更有效编码代码结构,提高注释生成的准确性,并提高代码处理模型的泛化能力。

Patent Agency Ranking