Skip to content

WangXuhongCN/APAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

APAN: Asynchronous Propagation Attention Network for Real-time Temporal Graph Embedding

Key words:

Graph Neural Network, Graph Embedding, Continuous-Time Dynamic Graph, Link Perdiction

Requirements:

pytorch>=1.5.0 DGL==0.5.2 numpy>=1.19 python >= 3.6

Dataset and Preprocessing

Download the public data

Download the datasets (eg. wikipedia and reddit) from here and store their csv files in a folder named data/. Note that the csv file should be renamed as wikipedia_raw.csv or reddit_raw.csv

Preprocess the data

Step 1. python preprocess/preprocess_csv.py --data wikipedia/reddit

Step 2. python preprocess/BuildDglGraph.py --data wikipedia/reddit

Model Training

python train.py -d wikipedia/reddit

General flags

Optional arguments are described in utils/args.py.

TODO

Cite us

Will be appeared at SIGMOD 2021

About

No description or website provided.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages