FeiGSSS / DySAT_pytorch

Pytorch implementation of DySAT: Deep Neural Representation Learning on Dynamic Graphs via Self-Attention Networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DySAT: Deep Neural Representation Learning on Dynamic Graphs via Self-Attention Networks

This is a pytorch implementation of DySAT. All codes are adapted from official implementation in TensorFlow. This implementation is only tested using dataset Enron, and the results is inconsistent with official results (better than that). Code review and contribution is welcome!

Raw Data Process

cd raw_data/Enron
pyhton process.py

The processed data will stored at 'data/Enron"

Training

python train --dataset Enron --time_steps 16

About

Pytorch implementation of DySAT: Deep Neural Representation Learning on Dynamic Graphs via Self-Attention Networks


Languages

Language:Python 100.0%