Skip to content

giannisnik/k-hop-gnns

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

k-hop Graph Neural Networks

Code for the paper k-hop Graph Neural Networks.

Requirements

Code is written in Python 3.6 and requires:

  • PyTorch 1.1
  • NetworkX 2
  • scikit-learn 0.21

Datasets

Use the following link to download datasets:

https://ls11-www.cs.tu-dortmund.de/staff/morris/graphkerneldatasets

Extract the datasets into the datasets folder.

Run the model

First, specify the dataset and the hyperparameters in the main.py file. Then, use the following command:

$ python main.py

Cite

Please cite our paper if you use this code:

@article{nikolentzos2020k,
  title={k-hop graph neural networks},
  author={Nikolentzos, Giannis and Dasoulas, George and Vazirgiannis, Michalis},
  journal={Neural Networks},
  volume={130},
  pages={195--205},
  year={2020},
  publisher={Elsevier}
}

Provided for academic use only

Releases

No releases published

Packages

No packages published

Languages