神经池:图神经网络的神经池 源码
图神经网络的神经池化 这是我的IAS SRFP 2020项目“图形神经网络的神经池”的代码。 Neural-Pooling文件夹包含代码。 Papers文件夹包含一些引用的论文。 Presentations文件夹包含每周的主题演讲。 系统要求 程式语言 Python 3.6 Python包 PyTorch > 1.0.0, tqdm, networkx, numpy 设置 如果您想尝试我们提出的神经池化方法,请复制graphcnn.py文件从以下任一位置进入Neural-Pooling/models文件夹Neural-Pooling/neural_pooling_1文件夹或Neural-Pooling/neural_pooling_2文件夹。 当前在Neural-Pooling/models文件夹中的graphcnn.py文件包含SUM / AVG池的代码。 运行代码 我们提供脚本来运
用户评论