Graph Representation Learning
2022-02-21 09:19:01 2.67MB graph machine learning
1
主要邻里聚集 在PyTorch,DGL和PyTorch Geometric中实现图网的主要邻域聚合 。 概述 我们提供PyTorch,DGL和PyTorch Geometric框架中的主要邻域聚合(PNA)的实现,以及用于生成和运行多任务基准的脚本,用于运行实际基准的脚本,灵活的PyTorch GNN框架以及其他实现用于比较的模型。 该存储库的组织方式如下: models包含: pytorch包含在PyTorch中实现的各种GNN模型: 聚合器,缩放器和PNA层( pna )的实现 可以与任何类型的图卷积一起使用的灵活GNN框架( gnn_framework.py ) 本文中用于比较的其他GNN模型的实现,即GCN,GAT,GIN和MPNN dgl包含通过实现的PNA模型:聚合器,缩放器和层。 pytorch_geometric包含通过实现的PNA模型:聚合器,缩放器和图层。
1