Graph classification research with graph convolutional networks in PyTorch
Top 83.3% on sourcepulse
This repository provides PyTorch implementations for graph classification using Graph Convolutional Networks (GCN), Graph U-Net, and Multigraph GCN (MGCN). It aims to reproduce and compare results from recent research papers, offering a baseline for evaluating graph neural network architectures on datasets like PROTEINS and ENZYMES.
How It Works
The core approach involves implementing graph convolutional layers and pooling mechanisms for graph classification. Graph U-Net utilizes a pooling strategy based on node dropping between graph convolution layers, differing from standard GCNs by simplifying the readout layer to max pooling. The implementation is basic, prioritizing clarity for debugging and experimentation over extensive optimization.
Quick Start & Requirements
pip
(PyTorch 0.4.1/1.0.0, Python 3.6 tested on Ubuntu 16.04).-g
flag.Highlighted Details
--degree
flag.Maintenance & Community
Licensing & Compatibility
Limitations & Caveats
The decoder part of Graph U-Net is not implemented. Performance may be affected by the basic implementation without optimizations. The project appears to be a research artifact from 2018, with no indication of ongoing maintenance or updates.
4 years ago
1 day