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

To run the code:

python batched_train.py

Name		Name	Last commit message	Last commit date
Latest commit History 35 Commits
.gitignore		.gitignore
MNIST.py		MNIST.py
README.md		README.md
batched_model.py		batched_model.py
batched_train.py		batched_train.py
dataset.py		dataset.py
utils.py		utils.py

Provide feedback