DGGAT

Dependencies

numpy
pandas
scikit-learn
scipy
torch==1.12.0+cu102
torch-geometric==2.1.0

Run DGGAT

python main.py -M cross_val -DM gate -O ./Out
python main.py -M train -DM gate -O ./Out
python main.py -M predict -DM gate -O ./Out --ModelPath ./model/model_gate.bin

Implementation of gating GAT

Our implementation of gating GAT is based on https://github.com/gordicaleksa/pytorch-GAT