← all repositories

benedekrozemberczki/ClusterGCN

A PyTorch implementation of Cluster-GCN, an algorithm for efficiently training large-scale graph convolutional networks on partitioned subgraphs.

806 stars Python ML FrameworksComputer Vision
ClusterGCN
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of the Cluster-GCN paper (KDD 2019), which uses graph clustering techniques to enable efficient SGD-based training of deep and large graph convolutional networks. It addresses the computational and memory challenges of training GCNs by sampling dense subgraphs identified through graph partitioning algorithms. The implementation includes the Amazon2M benchmark dataset with 2 million nodes and 61 million edges for scalability testing.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.