PyTorch implementation of Self-Attentive Sequential Recommendation

To train the model on MovieLens-1M (with default hyper-parameters):

python3 main.py

This implementation supports training with different hyperparameters. Refer to opts.py for more details. An example varying the attention stack (b=2), length of sequence (n=50) and hidden dimension (d=50):

python3 main.py --b=2 --n=50 --d=50 --n_epochs=10 --lr=0.001