graphsage simple:适用代码测试源代码地址:https 源码

参考PyTorch GraphSAGE实现 作者:威廉·汉密尔顿 基准PyTorch实施 。 此参考实现的速度不如大型图的TensorFlow版本快,但该代码更易于阅读,并且在小图基准测试中表现更好(在速度方面)。 与TensorFlow版本相比,该代码还旨在更简单,更可扩展且更易于使用。 当前,仅实现了GraphSAGE-mean和GraphSAGE-GCN的受监管版本。 要求 pytorch> 0.2是必需的。 运行示例 执行python -m graphsage.model来运行Cora示例。 它假定未使用CUDA,但是以明显的方式修改model.py中的运行函数可以改变这一点。 还有一个pubmed示例(通过run_pubmed中的run_pubmed函数调用)。