grad-june 0.1.0

Implementation of JUNE using Graph Neural Networks in PyTorch.

GradABM-JUNE

Implementation of the JUNE model using the GradABM framework.

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.13.0+cpu.html

Then install the GradABM-JUNE package

pip install -e .

Usage

See the docs.