[发明专利]目标代码的生成方法及装置

权利要求书

1.一种目标代码的生成方法，其特征在于，包括：

确定被测对象的代码标注信息；

将所述代码标注信息输入训练完成的生成模型，由所述生成模型输出对应的输出代码信息，其中，所述生成模型是先后通过第一数据集进行预训练，以及通过第二数据集进行调整训练得到的，所述第一数据集包括部分代码和对应的完整代码，所述第二数据集包括输入的代码标注信息以及对应的输出代码信息；

根据所述输出代码信息确定所述被测对象的目标代码，其中，所述目标代码用于对所述被测对象进行测试。

2.根据权利要求1所述的方法，其特征在于，将所述代码标注信息输入训练完成的生成模型，由所述生成模型输出对应的输出代码信息包括：

将所述代码标注信息输入所述生成模型的目标网络，由所述目标网络输出所述输出代码信息，其中，所述目标网络有多层，每层都设置有密集解码模块，相邻的密集解码模块之间设置有连接模块，多层密集解码模块按照解码捷径机制进行连接。

3.根据权利要求2所述的方法，其特征在于，将所述代码标注信息输入所述生成模型的目标网络，由所述目标网络输出所述输出代码信息包括：

将所述代码标注信息输入第一层的第一密集解码模块，由所述第一密集解码模块进行处理得到第一处理结果，并将处理结果发送给第二层的第二密集解码模块，以及后续的多个连接模块，其中，所述第一密集解码模块与所述第二密集解码模块之间直接连接；

通过所述第二密集解码模块将所述第一处理结果继续进行处理，得到第二处理结果，并将所述第二处理结果发送给第二层与第三层之间的第一连接模块，以及后续的其他连接模块，由所述第一连接模块将所述第二处理结果处理后发送给第三层的第三密集解码模块；

通过所述第三密集解码模块将所述第二处理结果继续进行处理，得到第三处理结果，并将所述第二处理结果发送给第三层与第四层之间的第二连接模块，以及后续的其他连接模块；

通过后续的密集解码模块进行处理，由最后一层的密集解码模块和最后一个连接模块输出所述输出代码信息。

4.根据权利要求3所述的方法，其特征在于，将所述代码标注信息输入第一层的第一密集解码模块，由所述第一密集解码模块进行处理得到第一处理结果包括：

将所述代码标注信息输入到所述第一密集解码模块的自注意力模块，由所述自注意力模块处理后得到第一解码信息；

将所述第一解码信息发送给第一切换正则化模块，得到第二解码信息，其中，所述自注意力模块与所述第一切换正则化模块之间为残差连接，所述第一切换正则化模块为层正则化函数和实例正则化函数组合确定的；

将所述第二解码信息发送给前馈模块，得到第三解码信息；

将所述第一解码信息，所述第二解码信息和所述第三解码信息输入第二切换正则化模块，由所述第二切换正则化模块输出所述第一处理结果。

5.根据权利要求1所述的方法，其特征在于，确定被测对象的代码标注信息包括：

确定所述被测对象的类文件；

提取所述类文件中的代码标注信息。

6.根据权利要求5所述的方法，其特征在于，将所述代码标注信息输入训练完成的生成模型，由所述生成模型输出对应的输出代码信息之前，所述方法还包括：

根据所述代码标注信息，生成预设数据格式的输入信息，其中，所述预设数据格式包括开始标志，结束标志以及所述代码标注信息；

根据所述输出代码信息确定所述被测对象的目标代码包括：

从所述预设数据格式的输出代码信息提取所述目标代码。

7.一种目标代码的生成模型的训练方法，其特征在于，包括：

获取第一数据集和第二数据集，其中，所述第一数据集和所述第二数据集的来源不同；

根据所述第一数据集对生成模型进行预训练，其中，所述第一数据集包括部分代码和对应的完整代码；

在预训练完成的情况下，根据所述第二数据集对所述生成模型进行调整训练，其中，所述第二数据集包括输入的代码标注信息以及对应的输出代码信息；

在所述调整训练验证通过的情况下，训练完成。