mirrors/HGSL
1
0
Fork 0
mirror of https://github.com/AndyJZhao/HGSL.git synced 2025-06-03 15:03:04 +08:00
Code Issues Projects Releases Wiki Activity
HGSL/README.md
Andy Zhao 25c0adf633
Update README.md
2021-06-07 23:23:09 +08:00

573 B
Raw Blame History

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'

Guide on Generation of 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.