mirrors
/
HGSL
mirror of https://github.com/AndyJZhao/HGSL.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Source code of AAAI21-Heterogeneous Graph Structure Learning for Graph Neural Networks
16 Commits
1 Branch
0 Tags
57 MiB
Python 100%
 
Go to file
Andy Zhao 1e39ba6c3d
Update README.md 		2022-08-03 22:41:09 +08:00
data Initial commit 2020-12-19 17:28:56 +08:00
paper Delete AAAI21_HGSL PrePrint.pdf 2021-04-20 15:56:46 +08:00
slides Slides updated 2021-02-23 23:01:49 +08:00
src Initial commit 2020-12-19 17:28:56 +08:00
.DS_Store Paper added 2020-12-19 22:26:40 +08:00
.gitignore Git ignore updated 2021-02-23 23:06:02 +08:00
LICENSE Initial commit 2020-12-08 11:23:53 +08:00
README.md Update README.md 2022-08-03 22:41:09 +08:00

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'

FAQ

Code of preprocessing data?

Please kindly note that the data is originally preprocessed by the GTN project (https://github.com/seongjunyun/Graph_Transformer_Networks).

I received quite a lot emails asking me about the dataset. I will not respond to them anymore as I cannot provide the code.

How to generate semantic embeddings?

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.