ivoryRabbit / hrnn-pytorch

Implementation of HRNN with Pytorch

HRNN-Pytorch

Paper

Personalizing Session-based Recommendations with Hierarchical Recurrent Neural Networks

Dataset

ml-1m

Clone

git clone [e-mail]:ivoryRabbit/hrnn-pytorch.git

Requirements

pip install -r requirements.txt

Load Data

python3 load_data.py

Preprocess

python3 preprocess.py --interaction ml-1m.csv

Training

python3 train.py --n_epochs 10 --batch_size 50

Recommend

python3 main.py --user_id 2 --eval_k 10

Reference

• HGRU4REC-Theano
• GRU4REC-PyTorch