Hierarchical Recurrent Neural Networks Pytorch version

Unofficial implementation of Personalizing Session-based Recommendations with Hierarchical Recurrent Neural Networks

work in progress

Dataset

• MovieLens 1M Dataset

cd {project Dir}/recomm-hrnn-pytorch/datasets/movielens
wget https://files.grouplens.org/datasets/movielens/ml-1m.zip
unzip ml-1m.zip

• MovieLens 10M Dataset

cd {project Dir}/recomm-hrnn-pytorch/datasets/movielens
wget https://files.grouplens.org/datasets/movielens/ml-10m.zip
unzip ml-10m.zip

Preprocess data

python preprocess.py -d 1M

Train model

example

python train_hrnn.py -d 1M -k 10 -lr 0.1 -v 0.0.1

Inference model

python inference.py -d 1M -w hrnn_v0.0.1_e03-loss0.2424.zip -k 10 --user 4452