摘要
本申请涉及一种氨基酸序列生成模型的训练方法及其应用。该方法包括:获取训练数据,训练数据包括样本蛋白质的样本结构信息;对样本结构信息进行特征编码,获得样本蛋白质对应的节点特征和边特征;通过待训练的图神经网络模型的多个消息传递层对节点特征和边特征进行交替更新,获得更新后的目标节点特征及目标边特征;根据对应的目标节点特征及目标边特征,通过待训练的图神经网络模型按照随机顺序分别对各节点进行解码预测,输出预测序列信息;根据预测序列信息与样本序列信息计算的损失值迭代图神经网络模型的网络参数,获得训练好的氨基酸序列生成模型。本申请提供的方案,能够提高生成模型预测结果的泛化性和鲁棒性。