mirror of https://github.com/AndyJZhao/HGSL.git
|
||
---|---|---|
data | ||
paper | ||
slides | ||
src | ||
.DS_Store | ||
.gitignore | ||
LICENSE | ||
README.md |
README.md
HGSL
Source code of AAAI submission "Heterogeneous Graph Structure Learning for Graph Neural Networks"
Requirements
Python Packages
- Python >= 3.6.8
- Pytorch >= 1.3.0
GPU Memmory Requirements
- ACM >= 8G
- DBLP >=5G
- Yelp >=3G
Usage
Take DBLP dataset as an example: python train.py --dataset='dblp'