WalkGNN

This is the official codebase of the paper

Ego-net Link Prediction with WalkGNN: A Scalable Approach for Large-Scale Social Graphs

Evgeny Zamyatin

The EgoVK dataset is avalilable here.

The Yeast dataset is avalilable here.

To reproduce the results, use the following command.

Train:

python3 train.py --model walk_gnn --device cuda:0

Validate:

python3 validate.py --model walk_gnn --device cuda:0 --state_dict_path <path>