Code for ICDM2021 Heterogeneous Graph Neural Network with Distance Encoding and its TKDE extended version: Distance Information Improves Heterogeneous Graph Neural Networks
- Python version >= 3.7
- PyTorch version >= 1.4.0
- torchmetrics version == 0.7.0
To run DHN on link prediction:
Python icdm.py
To run experiment with Efficient-HDE:
cd EHDE
python train.py
If you have any problems, feel free to contact me [email protected].