Usage

To use the code, enter the models directory and execute run_Model.py such as:

cd models/Caser
python run_Caser.py

SASRec: python main.py --dataset=ml-1m --train_dir=default --maxlen=200 --dropout_rate=0.2 --device=cuda

SSE-PT: python3 main.py --maxlen=200 --dropout_rate 0.2 --threshold_user 1 --threshold_item 1 --device=cuda

Note: Due to the different sample construction methods and experimental methods of different algorithms, we generate independent codes for each algorithm.

Requirements

• Tensorflow 1.1+
• Python 3.6+,
• numpy
• pandas

ToDo List

• More models
• Code refactoring
• Support tf.data.datasets and tf.estimator