Our article is PT-DGNN and the code is modified based on GPT-GNN
- python 3.6
- Ubuntu 18.04.5 LTS
- torch 1.4.0+cu100
- more packages are in requirements.txt
- pip install -r requirements.txt
- python pretrain.py --args values(args and values are detailed instructions in the code)
- python finetune.py --args values