rusty1s / pytorch_spline_conv

Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch
https://arxiv.org/abs/1711.08920
MIT License
172 stars 36 forks source link

Question about pooling #8

Closed siarez closed 5 years ago

siarez commented 5 years ago

Thanks for the implementation. It seems like the pooling that is described as: “The pooling operation is able to obtain a coarsened graph by deriving a clustering on the graph nodes, aggregating nodes in one cluster and computing new pseudo-coordinates ...” is actually not implemented in this repository. I'm trying to replicate this work, I'm wondering you computed the new pseudo-coordinates.

rusty1s commented 5 years ago

You can find all the additional code and examples here, e.g., in examples/mnist_graclus.py.