PTHGNN

source code of KDD 2021 paper "Pre-training on Large-Scale Heterogeneous Graph".

Environment Settings

Pytorch 1.4.0
pytorch_geometric 1.4.3
- torch-cluster==1.5.2
- torch-scatter==1.3.2
- torch-sparse==0.6.0
gensim
sklearn
tqdm
dill
pandas

Our datasets are uploaded to Baidu Netdisk

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
PT_HGNN		PT_HGNN
README.md		README.md
dataloader处理器.ipynb		dataloader处理器.ipynb
fe_finetune_AD.py		fe_finetune_AD.py
fe_finetune_PF.py		fe_finetune_PF.py
fe_finetune_PV.py		fe_finetune_PV.py
finetune_AD.py		finetune_AD.py
finetune_PF.py		finetune_PF.py
finetune_PV.py		finetune_PV.py
finetune_transfer_AD.py		finetune_transfer_AD.py
finetune_transfer_PF.py		finetune_transfer_PF.py
finetune_transfer_PV.py		finetune_transfer_PV.py
pretrain.py		pretrain.py
pretrain_structure.py		pretrain_structure.py