文件名称:神经池:图神经网络的神经池
文件大小:41.3MB
文件格式:ZIP
更新时间:2024-03-21 22:33:52
Python
图神经网络的神经池化 这是我的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池的代码。 运行代码 我们提供脚本来运