/SGINM

Code for paper "Learning Sequential And General Interests via A Joint Neural Model for Session-based Recommendation"

Primary LanguagePython

SGINM

Code for paper "Learning Sequential And General Interests via A Joint Neural Model for Session-based Recommendation (2020, Neurocomputing)"

How to Use

You directly run the file main.py to train the model.

Take Yoochoose 1/64 dataset as example:

python main.py --dataset yoochoose1_64 --batch_size 512

Requirements

  • Python3
  • pytorch==1.4.0

Citation

Please cite our paper if you use the code!