一种源代码漏洞预测方法、装置、电子设备和介质

本发明提供了一种源代码漏洞预测方法、装置、电子设备和介质，所述方法包括：获取待预测源代码数据集，对待测源代码数据集进行文本标记化处理得到多个源代码标记单元，根据源代码标记单元中的源代码文本建立词频排序词汇表，根据词频排序词汇表将源代码标记单元转换为独热编码格式的源代码特征向量，将独热编码格式的源代码特征向量输入源代码漏洞预测模型，输出漏洞类别预测结果。本发明通过对待预测源代码数据集的预处理以及通过训练好的源代码漏洞预测模型对经过预处理的待预测源代码数据集进行处理得到漏洞类别预测结果，实现对漏洞类别的准确预测，帮助开发人员和安全团队准确、高效地发现系统中的漏洞，减少潜在的安全风险。