PGCN:PGCN方法的PyTorch实现 源码
PGCN 渐进图卷积网络的PyTorch实现,用于半监督节点分类方法[1]。 这项工作是GCN方法[2]的扩展,并找到了GCN的优化网络架构。 预处理的Citeseer,Cora和Pubmed数据集可以在./pgcn/data目录中找到。 安装 python setup.py install 要求 PyTorch 0.4或0.5 Python 2.7或3.6 运行代码 git clone https://github.com/negarhdr/PGCN python setup.py install cd pgcn python pgcn.py --dataset cora --blocksize 5 --epochs 200 您可以在运行代码时指定模型的不同超参数。 参考 [1] [2] 致谢 此回购协议是基于修改的。 引用 如果您在自己的工作中使用以下代码,请引用我们
文件列表
PGCN-main.zip
(预估有个37文件)
PGCN-main
setup.py
573B
.gitignore
2KB
data
cora
normfeatures.npz
85KB
test_mask.npy
3KB
val_mask.npy
3KB
y_test.npy
148KB
train_mask.npy
3KB
暂无评论