Utilities & Misc
Training infrastructure and other examples that don't fit a single GNN task.
-
Distilling a GNN into an MLP (GLNN)
Graph-less neural networks (Zhang et al., 2021): train a GCN (the teacher), then train a plain MLP (the student) to reproduce the teacher's predictions from the node features alone.
Cora (Planetoid)·GCN · MLP -
Logging training to TensorBoard
Train the GCN of the Cora example and log the loss and accuracy of every epoch for TensorBoard.
Cora (Planetoid)·GCNConv -
Recommendation with LightGCN (MovieLens)
Recommend movies to users.
MovieLens100K·LightGCN -
Streaming data pipelines for graphs
PyG's example builds streaming pipelines with
torchdata: read raw files one by one, turn each record into a graph, and batch the graphs on the fly, without building a dataset on disk first.GeometricShapes·DataLoader