Graph Kernel Neural Networks

Install the following packages

numpy==1.24.4
scipy==1.10.1
scikit_learn==1.3.2
torch==1.9.0
torch-scatter==2.0.9 -f https://pytorch-geometric.com/whl/torch-1.9.0+cpu.html
torch-sparse==0.6.12 -f https://pytorch-geometric.com/whl/torch-1.9.0+cpu.html
torch_geometric==1.7.1
pytorch_lightning==1.1.6
grakel
wandb

Training

execute python run_training.py to start the training

Citation

@article{cosmo2021graph,
  title={Graph kernel neural networks},
  author={Cosmo, Luca and Minello, Giorgia and Bronstein, Michael and Rodol{\`a}, Emanuele and Rossi, Luca and Torsello, Andrea},
  journal={arXiv preprint arXiv:2112.07436},
  year={2021}
}