Pooling support for HeteroGraphs

CarloLucibello / GraphNeuralNetworks.jl

Graph Neural Networks in Julia

https://carlolucibello.github.io/GraphNeuralNetworks.jl/dev/

MIT License

209 stars 47 forks source link

Pooling support for HeteroGraphs #388

Open Ghaithq opened 4 months ago

Ghaithq commented 4 months ago

While browsing the codebase I noticed that there're some Convolution functions that support HeteroGraphs while there's none Pooling functions that support HeteroGraphs. So I thought it would be good to add this feature.

List of existing functions that need to be extended:

[ ] GlobalPool
[ ] GlobalAttentionPool
[ ] TopKPool
[ ] Set2Set

Ghaithq commented 4 months ago

@CarloLucibello I can try working on this if it's accepted as an issue.

CarloLucibello commented 4 months ago

Go for it!