Requirement:

• PyTorch > 1.4
• torch-geometric

Run the code:

python main.py --dataset (dataset_name) --setup (sid) [--prior_rate 0.02 --use_config (True/False) ...]

--datasets ppi/douban/arena/blogcatalog

Currently supports four datasets: PPI, Douban, Arena Emails, BlogCatalog

--setup 1/2/3/4

1 - GCN+Pseudo 2 - GAT+Pseudo 3 - LGCN+Pseudo 4 - LGCN+Wasserstein

--use_config True/False

if use_config is True (Default to be True), the config file in folder configs will be loaded.

--prior_rate

Only for Douban data, by default use 2% of the prior matrix.