二面角能量预测模型的训练方法和装置

本公开涉及人工智能技术领域，具体涉及二面角能量预测模型的训练方法和装置，方法公开了在训练二面角能量预测模型时，对用于训练的分子结构进行片段切分，得到二面角片段集合；基于二面角片段集合的第一片段子集，生成对应二面角片段样本集的第一样本子集；获取第一样本子集对应的特征矩阵集合的第一子集；采用二面角能量预测模型，根据特征矩阵集合的第一子集，生成对应特征矩阵集合的第一子集的第一预测能量值的集合，以及生成基于各个第一预测能量值与对应的标签能量值之间的差异确定的第一损失。本公开提升了二面角能量预测模型预测二面角片段的能量的准确性，并提升了训练完成的二面角能量预测模型的性能。