GND-Nets

Overview

Here we provide the implementation of GND-Nets (Graph Neural Diffusion Networks) in TensorFlow. The repository is organised as follows:

• data/ contains datasets Cora, Cora-ML, Citeseer, Pubmed, Amazon Computers, and Amazon Photo;
• new_data/ contains datasets Chameleon and Squirrel;
• models/ contains the implementation of the GND-Nets (gndnets_slp.py, gndnets_mlp.py, and gndnets_ds.py);
• utils/ contains:
  • an implementation of three variants of graph neural diffusions (layers.py);
  • preprocessing subroutines (process.py);

Finally, bash run_train execute the experiments.

Dependencies

The script has been tested running under Python 3.7.9, with TensorFlow version as:

• tensorflow==2.6.0

In addition, CUDA 11.4 and cuDNN 11.1 have been used.

License

MIT