Session based recommendation system using state of the art Transformers and Matrix Factorization on MovieLens-1M.
Please refer to the Report to view a detailed explanation of our project.
pip install -r requirements.txt
- Create
save/
directory inside the./Code
folder in order to save checkpoints - Run
python Code/train.py --max_seq_len 200 --num_layers 2
for the Transformer Model - Run
python Code/trainRNN.py --max_seq_len 200 --num_layers 2
for the RNN Model
tensorboard --logdir=runs