ptgnn:PyTorch图神经网络库 源码

ptgnn:PyTorch GNN库 这是一个包含pyTorch代码的库,用于创建图神经网络(GNN)模型。 该库提供了一些示例实现。 如果您对使用此库感兴趣,请阅读有关它的以及或阅读。 请注意, ptgnn负责定义整个管道,包括数据处理任务,例如数据加载和张紧。 它还为神经网络操作定义了PyTorch nn.Module 。 它们独立于AbstractNeuralModel并且可以用作所有其他PyTorch的nn.Module ,如果愿意的话。 该库主要设计用于快速处理稀疏图。 例如,对于Graph2Class任务上使用默认超参数和结构的V100(以下讨论) ptgnn可以处理约训练期间与约200图表/秒82个图表/秒(209K节点/秒和1,129k边缘/秒)(470K节点/秒和2,527k边/秒)。 已执行的任务 所有任务实现都可以在包中找到。 有关数据和培训步骤的详细说明,请