HeGCL

Source code for "HeGCL: Advance Self-Supervised Learning in Heterogeneous Graph-level Representation"

Important Dependencies

torch==1.7.1+cu101
dgl-cu101==0.6.1
h5py==3.5.0
networkx==2.3
ogb==1.3.1
pandas==1.2.5
scikit-learn==0.23.2

Example Data

You can download the data from the website:

https://drive.google.com/drive/folders/1-7Y7CNzmFsoz7F3uAnJ7rMlTrL_JmsU2?usp=sharing

Reference

If you make advantage of this paper in your research, please cite the following in your manuscript:

@ARTICLE{10135109,
  author={Shi, Gen and Zhu, Yifan and Liu, Jian K. and Li, Xuesong},
  journal={IEEE Transactions on Neural Networks and Learning Systems}, 
  title={HeGCL: Advance Self-Supervised Learning in Heterogeneous Graph-Level Representation}, 
  year={2023},
  volume={},
  number={},
  pages={1-12},
  doi={10.1109/TNNLS.2023.3273255}}

or

G. Shi, Y. Zhu, J. K. Liu and X. Li, "HeGCL: Advance Self-Supervised Learning in Heterogeneous Graph-Level Representation," in IEEE Transactions on Neural Networks and Learning Systems, doi: 10.1109/TNNLS.2023.3273255.

If you have any problem, please email me with this address [email protected].