A PyTorch Geometric implementation of the Greedy Total Resistance (GTR) rewiring algorithm. The accompanying paper can be found on Arxiv.

A Jupyter notebook with a simple example of how to use the gtr module can be found here.

The gtr module can be installed with pip using the following commands:

  cd src
  pip install .

gtr can also be uninstalled with the following commands:

  pip uninstall gtr