买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:蚂蚁云科技集团股份有限公司
摘要:本发明实施例提供了一种代码检索模型的训练方法、代码检索方法和装置,涉及数据处理技术领域,代码检索模型的训练方法,包括:获取目标代码库中的代码索引及其对应的相似索引;对代码索引进行分析得到对应的至少一个代码检索query信息;利用待训练的双塔模型分别对代码检索query信息、代码索引和相似索引进行编码,得到第一特征向量、第二特征向量和第三特征向量;计算第一特征向量与第二特征向量的第一相似度及第一特征向量与第三特征向量的第二相似度;基于第一相似度和第二相似度计算待训练的双塔模型的对比损失,并根据对比损失调整待训练的双塔模型的参数,得到训练好的代码检索模型。应该本发明实施例,提高了代码检索的准确率。
主权项:1.一种代码检索模型的训练方法,其特征在于,所述方法包括:获取目标代码库中的代码索引以及所述代码索引对应的相似索引;所述代码索引为所述目标代码库中代码块对应的信息,至少包括代码块对应的函数名和函数的注释信息;所述相似索引为与所述代码索引的相似度满足预设条件的代码块对应的信息;对所述代码索引进行分析,得到所述代码索引对应的至少一个代码检索查询query信息;所述代码检索query信息的数目与所述相似索引的数目相同;利用待训练的双塔模型对所述代码检索query信息进行编码得到第一特征向量、对所述代码索引进行编码得到第二特征向量、以及对所述相似索引进行编码得到第三特征向量;计算所述第一特征向量与所述第二特征向量之间的第一相似度,以及所述第一特征向量与所述第三特征向量之间的第二相似度;基于所述第一相似度和所述第二相似度,计算所述待训练的双塔模型的对比损失,并根据所述对比损失调整所述待训练的双塔模型的参数,得到训练好的代码检索模型。
全文数据:
权利要求:
百度查询: 蚂蚁云科技集团股份有限公司 一种代码检索模型的训练方法、代码检索方法和装置
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。