/diffpool

Primary LanguagePython

Differentiable Pooling PyTorch Implementation

This is not official implementaion.

Currently reach 60% validation accuracy without link prediction loss on Enzymes Dataset

Graph is batched for accelerating calculations

paper

To run the code:

python batched_train.py