HGSL/README.md

23 lines
869 B
Markdown
Raw Normal View History

2020-12-19 17:28:56 +08:00
# 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'
2021-06-07 23:23:09 +08:00
2021-06-07 23:28:33 +08:00
# FAQ
## Code of preprocessing data?
2022-05-17 07:26:50 +08:00
Please kindly note that the data is originally preprocessed by GTN project (https://github.com/seongjunyun/Graph_Transformer_Networks).
2022-05-17 07:27:09 +08:00
2022-05-17 07:26:50 +08:00
_I received quite a lot email asking me about the dataset. I will not respond to them anymore as I cannot provide the code._
2021-06-07 23:28:33 +08:00
## How to generate semantic embeddings?
2021-06-07 23:23:09 +08:00
The semantic embeddings, i.e. $\mathcal{Z}$ in the paper, are generated by metapath2vec algorithm. Users may refer to https://github.com/dmlc/dgl/tree/master/examples/pytorch/metapath2vec for an implementation.