Permute Me Softly: Learning Soft Permutations for Graph Representations
Code for the paper Permute Me Softly: Learning Soft Permutations for Graph Representations.
Requirements
Code is written in Python 3.7 and requires:
- PyTorch 1.9
- PyTorch Geometric 2.0
- NetworkX 2.2
Run the model
You can specify the dataset and the hyperparameters in the main.py file, and then execute:
python main.py
Cite
Please cite our paper if you use this code:
@article{nikolentzos2021permute,
title={Permute Me Softly: Learning Soft Permutations for Graph Representations},
author={Nikolentzos, Giannis and Dasoulas, George and Vazirgiannis, Michalis},
journal={arXiv preprint arXiv:2110.01872},
year={2021}
}
Provided for academic use only