Mobzhang / PT-DGNN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Pretraining on Dynamic Graph Neural Networks

Our article is PT-DGNN and the code is modified based on GPT-GNN

Requirements

  • python 3.6
  • Ubuntu 18.04.5 LTS
  • torch 1.4.0+cu100
  • more packages are in requirements.txt

Runing example

  1. pip install -r requirements.txt
  2. python pretrain.py --args values(args and values are detailed instructions in the code)
  3. python finetune.py --args values

About


Languages

Language:Python 100.0%