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!