[发明专利]反应物分子的预测、模型的训练方法、装置、设备及介质

权利要求书

1.一种反应物分子的预测方法，其特征在于，所述方法包括：

获取产物分子；基于所述产物分子，获取至少一个待补全分子；

调用目标分子补全模型对所述至少一个待补全分子进行补全，得到至少一个补全结果，基于所述至少一个补全结果获取所述产物分子对应的至少一个反应物分子；

其中，所述目标分子补全模型基于样本化合物分子以及样本待补全分子训练得到，所述样本待补全分子通过对所述样本化合物分子中的子结构进行掩码得到。

2.根据权利要求1所述的方法，其特征在于，所述调用目标分子补全模型对所述至少一个待补全分子进行补全，得到至少一个补全结果，包括：

对于所述至少一个待补全分子中的任一待补全分子，基于所述任一待补全分子的原子特征信息，获取所述任一待补全分子对应的补全后的分子的目标原子特征隐变量；基于所述任一待补全分子的化学键连接信息，获取所述任一待补全分子对应的补全后的分子的目标化学键连接隐变量；

调用所述目标分子补全模型对所述目标化学键连接隐变量进行变换，得到目标化学键连接信息；对所述目标原子特征隐变量进行变换，得到目标原子特征信息；

基于所述目标化学键连接信息和所述目标原子特征信息，获取所述任一待补全分子对应的补全结果。

3.根据权利要求1或2所述的方法，其特征在于，所述基于所述产物分子，获取至少一个待补全分子，包括：

获取所述产物分子的图结构信息；

基于所述图结构信息，预测所述产物分子中的化学键对应的断裂概率，将断裂概率满足参考条件的化学键作为所述产物分子中的断裂化学键；

基于所述断裂化学键对所述产物分子进行断键处理，得到所述至少一个待补全分子。

4.一种分子补全模型的训练方法，其特征在于，所述方法包括：

获取样本化合物分子以及样本待补全分子，所述样本待补全分子通过对所述样本化合物分子中的子结构进行掩码得到；

基于所述样本化合物分子、所述样本待补全分子和初始分子补全模型，获取训练损失；基于所述训练损失更新所述初始分子补全模型的模型参数，得到目标分子补全模型。

5.根据权利要求4所述的方法，其特征在于，所述基于所述样本化合物分子、所述样本待补全分子和初始分子补全模型，获取训练损失，包括：

获取所述样本化合物分子的样本原子特征信息和样本化学键连接信息；

基于所述样本化合物分子和所述样本待补全分子之间的差异，获取原子掩码信息和化学键掩码信息；

调用所述初始分子补全模型基于所述化学键掩码信息对所述样本化学键连接信息进行逆变换，得到样本化学键连接隐变量；基于所述原子掩码信息对所述样本原子特征信息进行逆变换，得到样本原子特征隐变量；

基于所述样本化学键连接隐向量和所述样本原子特征隐向量，获取所述训练损失。

6.根据权利要求5所述的方法，其特征在于，所述调用所述初始分子补全模型基于所述化学键掩码信息对所述样本化学键连接信息进行逆变换，得到样本化学键连接隐变量，包括：

调用所述初始分子补全模型基于所述化学键掩码信息和所述样本化学键连接信息，获取所述样本待补全分子的第一化学键连接信息以及所述子结构的第二化学键连接信息；

基于所述第一化学键连接信息对所述第二化学键连接信息进行逆变换，得到所述子结构的化学键连接隐变量；

基于所述第一化学键连接信息和所述子结构的化学键连接隐变量，获取所述样本化学键连接隐变量。

7.根据权利要求5所述的方法，其特征在于，所述基于所述原子掩码信息对所述样本原子特征信息进行逆变换，得到样本原子特征隐变量，包括：

基于所述原子掩码信息和所述样本原子特征信息，获取所述样本待补全分子的第一原子特征信息以及所述子结构的第二原子特征信息；

基于所述第一原子特征信息对所述第二原子特征信息进行逆变换，得到所述子结构的原子特征隐变量；

基于所述第一原子特征信息和所述子结构的原子特征隐变量，获取所述样本原子特征隐变量。