/NextItNet

PyTorch implementation of NextItNet from "A Simple Convolutional Generative Network for Next Item Recommendation".

Primary LanguagePython

NextItNet

PyTorch implementation of NextItNet from "A Simple Convolutional Generative Network for Next Item Recommendation".

Use the leave-one-out strategy to train, validate, and test network parameters.

Dataset: MovieLens-1M https://grouplens.org/datasets/movielens/1m/

Changed ReLU activation function to leaky-ReLU from original paper.