科研之友
微信
新浪微博
Facebook
分享链接
摘要
源代码检索任务是指将自然语言作为查询语句，从代码库中搜索相关代码片段。在代码检索任务中，大多数代码检索算法只考虑代码片段的文本序列信息而未考虑代码的结构信息，导致不能充分捕获代码片段包含的语义和语法信息。为了提高对程序语言的理解，提出了注意力机制和图嵌入相结合的代码检索算法（GraphCS）。在特征提取部分，以LSTM来提取文本特征向量表示，以Graph2Vec来提取图的向量特征表示。在特征融合部分中引入注意力机制，更好地为每一个特征分配相应的权重，从而提升程序的理解。考虑源代码和自然语言为异构数据，将代码片段特征和自然语言特征映射到同一个向量空间，以排名损失来保证语义相似的点在特征空间拥有较近的距离。为了验证算法的高效性，与目前最好的算法CODEnn进行对比。实验结果表明，在Precision@1/5/10，SuccessRate@1/5/10以及MRR上均有一定的提升。
- 单位
