
GraphSAGE-Simple: 适用于代码测试的源代码地址:https
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
GraphSAGE-Simple是一款简洁高效的图神经网络模型实现工具,专注于简化代码结构以促进软件测试。项目源码位于GitHub,欢迎贡献与反馈。
参考PyTorch GraphSAGE实现作者是威廉·汉密尔顿的基准PyTorch实施版本。此参考实现的速度在处理大型图时不如TensorFlow版本快,但在小图基准测试中表现更好(特别是在速度方面)。相较于TensorFlow版本,该代码旨在更简单、更可扩展且易于使用。目前仅实现了GraphSAGE-mean和GraphSAGE-GCN的受监管版本。
要求:pytorch> 0.2是必需的。
运行示例:
执行 `python -m graphsage.model` 来运行Cora示例,它假定未使用CUDA,但可以通过修改model.py中的run函数来改变这一点。还有一个pubmed示例(通过调用run_pubmed中的run_pubmed函数)。
全部评论 (0)
还没有任何评论哟~


