高速公路:高速公路的存储库:基于自同构的图神经网络 源码

概述 该存储库代码用于论文《高速公路:基于自同构的图神经网络》 。 与大多数图神经网络(其中神经元与单个图顶点相关联)相反,高速公路中的神经元对应于子图。 然后,每个神经元都应用与每个子图的同构组等价的卷积,然后是固定的非线性。 确保神经元与局部自同构群是等变的,足以使整个网络与输入的排列是等变的。 在我们的代码中,我们构建了一个高速公路网络,其中神经元对应于每个图中的路径和循环。 产生的神经元反映了我们对化学图的领域知识,并保留了在顺序数据上进行卷积的直观含义。 免费软件:MIT许可证 安装 克隆存储库并运行 pip install -e . 培训网络 培训脚本位于src/autobahn/experiments文件夹中。 训练脚本会自动尝试下载所需的数据集并运行模型。 python -um autobahn.experiments.train_combo_on_zinc