基于注意力机制的图分类方法及化合物毒性预测方法

摘要

本发明公开了一种基于注意力机制的图分类方法，包括获取原始图网络及对应的图网络数据信息；初始化节点的特征嵌入、构建高阶图网络并初始化高阶图网络中的节点嵌入；各阶图网络中的节点采用子结构交互注意力网络捕获吸收邻居节点信息和邻居节点间的交互信息并训练得到图网络中每个节点的新的特征嵌入；采用节点注意力网络将图网络中的节点特征嵌入融合成图网络嵌入，级联各阶图网络嵌入后通过多层感知机将降维分类并得到最终的图网络分类结果。本发明还公开了一种包括所述基于注意力机制的图分类方法的化合物毒性预测方法。本发明有效提高了图网络分类的准确度和化合物毒性预测的准确度；而且效率较高、准确性较好且易于实施。