search

lukecavabarrett / pna

Implementation of Principal Neighbourhood Aggregation for Graph Neural Networks in PyTorch, DGL and PyTorch Geometric
https://arxiv.org/abs/2004.05718
MIT License
338 stars 55 forks source link

The use of the pytorch code #13

Closed yangyang117 closed 2 years ago

yangyang117 commented 2 years ago

Hi, thank you very much for your sharing of your code. Is there any way that in which kind of code, you can provide a examples how to create the input and how to use the layer?

gcorso commented 2 years ago

Hi, For the DGL version of the code you can look at all of our experiments and the docs of DGL for a much simpler one. For PyTorch Geometric you can find here a relatively simple example. I hope this helps!