jiangqn / GCN-GAN-pytorch

A pytorch implemention of GCN-GAN for temporal link prediction.

GCN-GAN

This repository provide a pytorch implemention for the GCN-GAN model proposed in "A Non-linear Temporal Link Prediction Model for Weighted Dynamic Networks" INFOCOM 2019, [pdf].

Quick Start

Modify hyper-parameters in file config.yml.

Preprocess data

python preprocess.py

Train

python train.py

Test

python test.py