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

GitHub

View Github