文件名称:PGCN:PGCN方法的PyTorch实现
文件大小:6.17MB
文件格式:ZIP
更新时间:2024-03-28 13:45:14
Python
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
----setup.py(573B)
----.gitignore(2KB)
----data()
--------cora()
--------pubmed()
--------citeseer()
----pgcn()
--------__init__.py(136B)
--------layers.py(1KB)
--------models.py(860B)
--------utils.py(626B)
--------pgcn.py(11KB)
----LICENSE(1KB)
----README.md(1KB)
----Model-Diagram.png(925KB)