ag027592 / HGSL

Source code of AAAI21-Heterogeneous Graph Structure Learning for Graph Neural Networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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'

About

Source code of AAAI21-Heterogeneous Graph Structure Learning for Graph Neural Networks

License:Apache License 2.0


Languages

Language:Python 100.0%