pyg-team/pytorch_geometric - GitHub
https://github.com/pyg-team/pytorch_geometric
OverviewLibrary HighlightsQuick Tour for New UsersArchitecture OverviewImplemented GNN ModelsInstallationCiteTrain your own GNN modelIn the first glimpse of PyG, we implement the training of a GNN for classifying papers in a citation graph. For this, we load the Cora dataset, and create a simple 2-layer GCN model using the pre-defined GCNConv:We can now optimize the model in a training loop, similar to the standard PyTorch training proc…More information about evaluating final model performance can be found in the corresponding example.See more on github.comExplore further Train your own GNN modelIn the first glimpse of PyG, we implement the training of a GNN for classifying papers in a citation graph. For this, we load the Cora dataset, and create a simple 2-layer GCN model using the pre-defined GCNConv: We can now optimize the model in a training loop, similar to the standard PyTorch training proc…More information about evaluating final model performance can be found in the corresponding example.
Train your own GNN modelIn the first glimpse of PyG, we implement the training of a GNN for classifying papers in a citation graph. For this, we load the Cora dataset, and create a simple 2-layer GCN model using the pre-defined GCNConv:
We can now optimize the model in a training loop, similar to the standard PyTorch training proc…More information about evaluating final model performance can be found in the corresponding example.
DA: 71 PA: 4 MOZ Rank: 98