Skip to content
/ DGSR Public
forked from CRIPAC-DIG/DGSR

[TKDE 2022] The source code of "Dynamic Graph Neural Networks for Sequential Recommendation"

Notifications You must be signed in to change notification settings

Alvin0412/DGSR

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

56 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DGSR

model

This is the code for the TKDE 2022 Paper: Dynamic Graph Neural Networks for Sequential Recommendation.

Usage

Generate data

You need to run the file new_data.py to generate the data format needed for our model. The detailed commands can be found in load_{dataset}.sh

You need to run the file generate_neg.py to generate necessary negative samples. You can set the data set in the file.

Training and Testing

Then you can run the file new_main.py to train and test our model. The detailed commands can be found in {dataset}.sh

Requirements

  • Python 3.6
  • torch 1.7.1
  • dgl 0.7.2

Citation

Please cite our paper if you use the code:

@ARTICLE{9714053,
  author={Zhang, Mengqi and Wu, Shu and Yu, Xueli and Liu, Qiang and Wang, Liang},
  journal={IEEE Transactions on Knowledge and Data Engineering}, 
  title={Dynamic Graph Neural Networks for Sequential Recommendation}, 
  year={2022},
  volume={},
  number={},
  pages={1-1},
  doi={10.1109/TKDE.2022.3151618}}

About

[TKDE 2022] The source code of "Dynamic Graph Neural Networks for Sequential Recommendation"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 98.1%
  • Shell 1.5%
  • PowerShell 0.4%