WalkGNN

This is the official codebase of the paper

GNN Applied to Ego-nets for Friend Suggestions

Evgeny Zamyatin

Dataset

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>

Name		Name	Last commit message	Last commit date
Latest commit History 106 Commits
models		models
.gitignore		.gitignore
README.md		README.md
dataset.py		dataset.py
loss.py		loss.py
requirements.txt		requirements.txt
train.py		train.py
train_YEAST.py		train_YEAST.py
validate.py		validate.py