相互作用蛋白质生成模型的训练方法、装置

本发明提供一种相互作用蛋白质生成模型的训练方法、装置，方法包括：获取样本蛋白质对，所述样本蛋白质对包含样本蛋白质，和所述样本蛋白质的样本相互作用蛋白质；提取并融合所述样本蛋白质对中各氨基酸残基的序列特征和空间结构特征，得到所述样本蛋白质对的残基特征；基于初始生成模型和所述残基特征，生成所述样本蛋白质的样本预测蛋白质；基于所述样本预测蛋白质构建生成损失，基于所述生成损失对所述初始生成模型进行参数迭代，得到相互作用蛋白质生成模型。本发明提供的方法，通过在训练时同时考虑蛋白质的一阶序列信息和三维结构信息，大大提升了生成相互作用蛋白质的准确性，实现了高效、通用且具有亲和力的相互作用蛋白质生成。