GraphNeuralNetworks.jl
GraphNeuralNetworks.jl copied to clipboard
Implement more pooling operators
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
what's the status of this now?
I'm not actively working on pooling operators right now
I read doc of pool: GraphNeuralNetworks.GlobalAttentionPool GraphNeuralNetworks.GlobalPool GraphNeuralNetworks.TopKPool
Doesn't that already exist?
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.