Self-supervised Heterogeneous Graph Learning: a Homophily and Heterogeneity View

This repository contains the reference code for the manuscript ``Self-supervised Heterogeneous Graph Learning: a Homophily and Heterogeneity View"

Installation

pip install -r requirements.txt
Unzip the datasets (the heterogeneous graph datasets can be found in ./dataset/, and the homogeneous graph datasets will be automatically downloaded from the public website)

Important args:

--use_pretrain Test checkpoints to reproduce the results
--dataset Heterogeneous graph: ACM, Yelp, DBLP, Aminer || Homogeneous graph: photo, computers, cs, physics
--custom_key Node: node classification

python main.py

Name		Name	Last commit message	Last commit date
Latest commit History 15 Commits
dataset		dataset
layers		layers
models		models
saved_model		saved_model
utils		utils
README.md		README.md
__init__.py		__init__.py
args.yaml		args.yaml
embedder.py		embedder.py
evaluation.py		evaluation.py
main.py		main.py
requirements.txt		requirements.txt