qiaoyu-tan/CONN

CONN

This is the official pytorch implementation of Collaborative Graph Neural Networks for Attributed Network Embedding.

Installation

The required packages can be installed by running pip install -r requirements.txt.

Collaborative Graph Neural Networks

Datasets

The datasets used in our paper can be automatically downlowad.

Quick Start

Test on node classification task:

python main_classify.py --dataset "PubMed" 

Test on link prediction task:

python main_lp.py --dataset "PubMed"