GraphNeuralNetworks.jl icon indicating copy to clipboard operation
GraphNeuralNetworks.jl copied to clipboard

Implement more pooling operators

Open CarloLucibello opened this issue 3 years ago • 4 comments

This is the list of pooling operators in pytorch geometric

  • [x] global_add_pool (GlobalPool(+) here)
  • [x] global_mean_pool (GlobalPool(mean) here)
  • [x] global_max_pool (GlobalPool(max) here)
  • [ ] global_sort_pool
  • [x] GlobalAttention (GlobaAttentionlPool(max) here)
  • [ ] Set2Set
  • [ ] GraphMultisetTransformer

CarloLucibello avatar Feb 12 '22 10:02 CarloLucibello

what's the status of this now?

zsz00 avatar Feb 25 '22 12:02 zsz00

I'm not actively working on pooling operators right now

CarloLucibello avatar Feb 25 '22 12:02 CarloLucibello

I updated the list with the current status. TopK I think is broken and has to be completely reworked. Also, we don't have sort pooling and set2set.

CarloLucibello avatar Feb 25 '22 13:02 CarloLucibello