摘要
本发明公开了一种基于迁移学习的自动化软件重构方法,包括:步骤一、利用代码数据集训练基于BERT的标签预测模型,计算出的辅助任务软标签及重构任务软标签,构建标签映射矩阵;步骤二、根据节点类型和边类型对所述代码数据集进行处理,构建代码重构结构图;采用图神经网络提取代码结构信息,采用Code2Vec提取代码语义信息,将所述代码结构信息与所述代码语义信息融合生成混合向量;将所述混合向量输入全连接层,利用激活函数生成重构任务标签及辅助任务标签;将所述辅助任务标签输入所述标签映射矩阵生成重构映射标签;步骤三、将所述重构任务标签、所述重构映射标签及所述混合向量输入到基于特征迁移的Transformer模型中,得到重构后代码。