[发明专利]代码推荐方法和装置

说明书

本申请实施例提供了一种代码推荐方法、装置、电子设备和计算机可读存储介质，解决了现有代码自动推荐功能无法给用户提供类级别的参考，且代码推荐效率不高的问题。该代码推荐方法包括：将多个待选数据输入训练好的分类模型，得到与多个所述待选数据一一对应的多个分类，其中，多个所述待选数据用于表征多个候选代码的内容；根据与多个所述待选数据一一对应的多个所述分类，获取与多个所述分类一一对应的多个分类相似度值；以及根据所述多个分类相似度值，从所述多个候选代码中获取推荐代码。

技术领域

本发明涉及代码推荐技术领域，具体涉及一种代码推荐方法、装置、电子设备和计算机可读存储介质。

背景技术

现代软件开发工作流程的主体是集成开发环境(IDE)，许多高级开发工具的IDE的核心功能中包括了代码自动推荐功能。然而IDE中包含的代码推荐系统只能实现应用程序接口(API)级别的推荐，即只能实现代码补全，因此，只是免去了输入冗长单词的操作，无法给用户提供类级别的参考，且代码推荐效率不高。

发明内容

有鉴于此，本发明实施例提供了一种代码推荐方法、装置、电子设备和计算机可读存储介质，解决了现有代码自动推荐功能无法给用户提供类级别的参考，且代码推荐效率不高的问题。

根据本申请的一个方面，本申请一实施例提供一种代码推荐方法，包括：将多个待选数据输入训练好的分类模型，得到与多个所述待选数据一一对应的多个分类，其中，多个所述待选数据用于表征多个候选代码和用户代码的内容；根据与多个所述待选数据一一对应的多个所述分类，获取与多个所述分类一一对应的多个分类相似度值；以及根据所述多个分类相似度值，从所述多个候选代码中获取推荐代码。

在本申请一实施例中，所述根据所述多个分类相似度值，从所述多个候选代码中获取推荐代码包括：根据所述多个分类相似度值和初步相似度值的和，从所述多个候选代码中获取推荐代码；其中，所述根据所述多个分类相似度值和初步相似度值的和，从所述多个候选代码中获取推荐代码之前还包括：根据多个所述候选代码和所述用户代码，得到与多个所述候选代码一一对应的多个初步相似度值；以及将与多个所述候选代码一一对应的多个所述初步相似度值和多个所述分类相似度值分别相加，得到与多个所述候选代码一一对应的所述多个分类相似度值和初步相似度值的和。

在本申请一实施例中，所述将多个待选数据输入训练好的分类模型之前还包括：对多个原始候选代码进行特征提取，得到与所述多个原始候选代码对应的所述多个原始候选代码的特征矩阵；对所述用户代码进行特征提取，得到与所述用户代码对应的所述用户代码的特征向量；将所述多个原始候选代码的所述特征矩阵与所述用户代码的所述特征向量相乘，得到所述多个原始候选代码与所述用户代码的初步相似度向量，其中，所述相似度向量包含与所述多个原始候选代码一一对应的多个相似度向量元素值；以及提取与所述多个原始候选代码一一对应的多个所述相似度向量元素值大于预设阈值的代码，得到多个候选代码。

在本申请一实施例中，所述根据多个所述候选代码和所述用户代码，得到与多个所述候选代码一一对应的多个初步相似度值包括：提取与多个所述候选代码一一对应的多个相似度向量元素值；以及；以及将与多个所述候选代码一一对应的多个所述相似度向量元素值进行归一化处理，得到与多个所述候选代码一一对应的多个所述初步相似度值。

在本申请一实施例中，所述根据所述多个候选代码和所述用户代码，得到与多个所述候选代码一一对应的多个初步相似度值之前还包括：在项目集平台收集多个项目；以及提取多个所述项目中每个所述项目的多个类文件，得到所述多个原始候选代码。

在本申请一实施例中，所述在项目集平台收集多个项目包括：在项目集平台收集多个关注度大于预设数值的项目。