文件名称:benchmarking-gnns-pyg
文件大小:2.25MB
文件格式:ZIP
更新时间:2024-06-16 06:21:20
JupyterNotebook
基准图神经网络用于节点分类 注意:此存储库是从graphdeeplearning / benchmarking-gnns修改的。 我们使用pyg框架重构代码,并添加Planetoid和ogb数据集以进行节点分类。 1.基准安装 为GPU设置Python环境 git clone https://https://github.com/karl-zhao/benchmarking-gnns-pyg.git cd benchmarking-gnns-pyg # Install python environment conda env create -f environment_gpu.yml # Activate environment conda activate pytorch1.5.0 2.下载数据集 除SBM外,所有数据集均可自动下载。 对于SBM,首先运行和以生成数据集。 3.重现