pytorch_geometric
Graph Neural Network Library for PyTorch. 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 procedure .
What it solves
Graph Neural Network Library for PyTorch. 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 procedure .
The project context on this page is free to read. The original GitHub repository remains the source of truth; sign in only when you want to save or join the discussion.
Community notes