mirror of https://github.com/AndyJZhao/HGSL.git
14 lines
316 B
Markdown
14 lines
316 B
Markdown
# 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'
|