一种代码查询模型的生成方法和计算机设备

摘要

本发明提供了一种代码查询模型的生成方法和计算机设备，代码查询模型的生成方法包括：在训练集中选取匹配语句对和非匹配语句对；基于初始词嵌入模块匹配词嵌入向量集和非匹配向量集，基于初始Transformer模块确定匹配词嵌入矩阵集以及非匹配词嵌入矩阵集；基于初始交互注意力模块确定第一相似度和第二相似度；基于第一相似度和第二相似度训练得到代码查询模型。本发明通过初始词嵌入模块和初始Transformer模块学习查询语句和代码语句多层面语义信息，使得训练得到的代码查询模型可以在程序代码数据库庞大、结构繁杂时准确匹配查询语句对应的匹配代码语句。