一种基于细粒度共注意机制的语义代码搜索方法

摘要

本发明涉及一种基于细粒度共注意机制的语义代码搜索方法，包括如下步骤：首先，将代码特征(即方法名、API序列、令牌和语句树序列)和查询特征(即令牌序列)输入到一个注意力机制中，来获得代码/查询特征矩阵；然后，将获得的代码/查询特征矩阵分别输入四种细粒度的共注意机制中，得到每个代码特征与查询之间的细粒度相关性；最后，相互依赖的特征向量分别融合成最终的代码/查询向量，并计算两个代表性向量之间的余弦相似性，将余弦相似性降序排列，得到最终推荐的N个查询结果。使用本方法可以增强对代码/查询特征向量的信息使用效率，提高代码查询最终结果的准确性。