https://github.com/vuptran/node-classification
Large-Scale Node Classification with PyTorch Geometric and Lightning
https://github.com/vuptran/node-classification
deep-learning graph-neural-networks node-classification node-embedding pytorch-geometric pytorch-lightning
Last synced: 11 days ago
JSON representation
Large-Scale Node Classification with PyTorch Geometric and Lightning
- Host: GitHub
- URL: https://github.com/vuptran/node-classification
- Owner: vuptran
- License: mit
- Created: 2025-11-23T05:45:21.000Z (2 months ago)
- Default Branch: main
- Last Pushed: 2025-12-23T03:35:43.000Z (about 1 month ago)
- Last Synced: 2025-12-24T18:11:37.328Z (about 1 month ago)
- Topics: deep-learning, graph-neural-networks, node-classification, node-embedding, pytorch-geometric, pytorch-lightning
- Language: Python
- Homepage:
- Size: 27.3 KB
- Stars: 0
- Watchers: 0
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# 📘 Large-Scale Node Classification
Scalable node classification on millions of nodes and edges by leveraging PyTorch Geometric and Lightning. Train on a single GPU or multiple GPUs with benchmarks from the [OGB project](https://ogb.stanford.edu/docs/nodeprop/) using classical Graph Neural Networks like GraphSAGE and GAT.
## 🚀 Get Started
1. Refer to this [installation guide](https://github.com/vuptran/node-classification/tree/main/docs/installation).
2. Download and convert the [supported datasets](https://github.com/vuptran/node-classification/tree/main/data).
3. Run the example [training scripts](https://github.com/vuptran/node-classification/tree/main/scripts).
## ❤️ Acknowledgments
We are grateful for the open-source contributions from
* [PyTorch Lightning](https://lightning.ai/docs/pytorch/stable/)
* [PyTorch Geometric](https://github.com/pyg-team/pytorch_geometric)
* [tunedGNN [NeurIPS'24]](https://github.com/LUOyk1999/tunedGNN)