This repo contains the code used for HetSAGE: Heterogenous Graph Neural Network for Relational Learning.
Use the environment.yml file to install the required packages.
To reproduce the results from the paper:
- preprocess the data:
python sql2graph.py --opts <opts>, e.g.--opts rules/muta188.yml, - run the training:
python exp.py --graph <graph-file> --max-epochs 50 --workers 2 --logdir <logdir> --device cpu