Torch JUNE

Implementation of the JUNE model using Graph Neural Networks in PyTorch

Setup

Install requirements

pip install -r requirements.txt

and install PyTorch geometric, manually for now:

pip install torch-scatter torch-sparse torch-cluster torch-geometric -f https://data.pyg.org/whl/torch-1.11.0+cpu.html

Then install the torch_june package

pip install --no-deps -e .

Usage

See the docs.