CarloLucibello / GraphNeuralNetworks.jl

Graph Neural Networks in Julia
https://carlolucibello.github.io/GraphNeuralNetworks.jl/dev/
MIT License
220 stars 46 forks source link

Add GINConv support to HeteroGraphConv #387

Closed rbSparky closed 7 months ago

rbSparky commented 7 months ago

Covers Issue #311

Need some advice on how to integrate this in the tests, please let me know. For now I have only written a dummy test since I wasn't sure how to test GINConv the right way.

CarloLucibello commented 7 months ago

you can add a simple test of the output shape, as for the other tests in that file. Then this seems good to go.