Torch Geometric Pool: the PyTorch library for pooling in Graph Neural Networks
About
Torch Geometric Pool (tgp) is a pooling library built on top of PyTorch Geometric. Graph pooling methods differ in how they assign nodes to supernodes, how they handle batches, what they return after pooling, and whether they expose auxiliary losses. These differences make it hard to compare methods or reuse the same model code across them. tgp addresses this problem with a common software interface based on the Select-Reduce-Connect-Lift (SRCL) decomposition. The library provides 20 hierarchical poolers, standardized output objects, standalone readout modules, support for dense poolers in batched and unbatched mode, and workflows for caching and pre-coarsening. It is released under the MIT license on GitHub and PyPI, with comprehensive documentation, tutorials, and examples.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Graph Classification | NCI1 | -- | 707 | |
| Node Classification | amazon-ratings | -- | 354 | |
| Node Clustering | Cora | -- | 179 | |
| Graph Classification | REDDIT-B | -- | 163 | |
| Node Classification | questions | -- | 161 | |
| Graph Regression | Peptides-struct | -- | 156 | |
| Node Clustering | Citeseer | -- | 151 | |
| Graph Classification | Peptides func | -- | 132 | |
| Node Classification | Minesweeper | -- | 117 | |
| Graph Classification | MolHIV | -- | 110 |