[发明专利]一种分子生成方法和计算设备

权利要求书

1.一种分子生成方法，适于在计算设备中执行，所述计算设备中存储有分子片段库，所述分子片段库包括从多个已知分子中拆分出的多个分子片段，每个分子片段具有向量表示，所述方法包括步骤：

生成一个多维向量作为一个生成分子的树向量，所述树向量适于代表该生成分子的联结树，所述联结树的节点为由所述生成分子拆分出的多个分子片段，相邻的分子片段用边连接；

从作为全零向量的空节点开始，采用预设的解码模型依次判断每个当前节点之后是否需要新增下一节点，并在需要新增时从分子片段库中选取下一节点所对应的分子片段，从而得到所述联结树；以及

确定所述联结树中相邻分子片段之间的连接方式，并采用该连接方式对相邻分子片段进行连接，得到所述生成分子。

2.如权利要求1所述的方法，其中，所述生成一个多维向量作为树向量的步骤包括：

根据预设的树向量维度和分布特点随机生成一个对应维度的向量作为所述生成分子的树向量。

3.如权利要求1所述的方法，其中，所述生成一个多维向量作为树向量的步骤包括：

接收用户的输入分子，将该输入分子拆分为多个分子片段；

将其中一个分子片段作为根节点，并根据各分子片段之间的连接关系生成所述输入分子的联结树；

采用预设的编码模型对所述输入分子的联结树进行编码，得到对应的树向量；

对所述输入分子的树向量叠加微小扰动后，得到所述生成分子的树向量。

4.如权利要求1-3中任一项所述的方法，其中，

所述编码模型的输入为第一组合向量、输出为从当前节点指向下一节点的出链边向量，所述第一组合向量包括当前节点向量和指向当前节点的入链边向量，所述当前节点的入链边向量为上一节点的出链边向量。

5.如权利要求4所述的方法，其中，所述采用预设的编码模型对所述输入分子的联结树进行编码的步骤包括；

从该联结树的叶子节点开始，基于所述编码模型向根节点方向依次计算每个节点的出链边向量，以得到所述根节点的入链边向量；以及

将根节点向量和所述根节点的入链边向量进行拼接，并经过全连接层降维后，得到所述输入分子的树向量。

6.如权利要求5所述的方法，其中，若所述根节点有多个入链边向量，则所述将根节点向量和所述根节点的入链边向量进行拼接的步骤包括：

将该多个入链边向量进行求和之后，与所述根节点向量进行拼接。

7.如权利要求1-6中任一项所述的方法，其中，

所述解码模型的输入为第二组合向量、输出为当前节点之后是否新增子节点，所述第二组合向量包括当前节点向量、指向当前节点的入链边向量之和、以及所述生成分子的树向量；

其中，初始的当前节点为空节点，其当前节点向量和指向当前节点的入链边向量均为全零向量。

8.如权利要求4所述的方法，其中，所述从分子片段库中选取下一节点所对应的分子片段的步骤包括：

基于当前节点向量和指向当前节点的入链边向量，采用所述编码模型得到当前节点的出链边向量；

将所述生成分子的树向量和当前节点的出链边向量进行拼接，并经过全连接层降维后，计算每个降维后的向量与所述分子片段库中的节点向量的相似度；

将相似度最高的片段作为下一节点所对应的分子片段，从而得到所述生成分子的联结树。

9.一种计算设备，包括：

存储器；

一个或多个处理器；

一个或多个程序，其中所述一个或多个程序存储在所述存储器中并被配置为由所述一个或多个处理器执行，所述一个或多个程序包括用于执行根据权利要求1-8所述方法中的任一方法的指令。

10.一种存储一个或多个程序的计算机可读存储介质，所述一个或多个程序包括指令，所述指令当计算设备执行时，使得所述计算设备执行根据权利要求1-8所述的方法中的任一方法。