Skip to content
forked from CCIIPLab/GCE-GNN

The source code for "Global Context Enhanced Graph Neural Network for Session-based Recommendation".

Notifications You must be signed in to change notification settings

caiyongqi/GCE-GNN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GCE-GNN

Code

This is the source code for SIGIR 2020 Paper: Global Context Enhanced Graph Neural Networks for Session-based Recommendation.

Requirements

  • Python 3
  • PyTorch >= 1.3.0
  • tqdm

Usage

Data preprocessing:

The code for data preprocessing can refer to SR-GNN.

Train and evaluate the model:

python build_graph.py --dataset diginetica --sample_num 12
python main.py --dataset diginetica

Citation

@inproceedings{wang2020global,
    title={Global Context Enhanced Graph Neural Networks for Session-based Recommendation},
    author={Wang, Ziyang and Wei, Wei and Cong, Gao and Li, Xiao-Li and Mao, Xian-Ling and Qiu, Minghui},
    booktitle={Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval},
    pages={169--178},
    year={2020}
}

About

The source code for "Global Context Enhanced Graph Neural Network for Session-based Recommendation".

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%