pyg-team
★ 23,960pytorch_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 .