session4rec

GRu4Rec in Tensorflow

Get data

• http://2015.recsyschallenge.com/challenge.html
• https://s3-eu-west-1.amazonaws.com/yc-rdata/yoochoose-data.7z

Preprocess data

• change input and output directories in /scripts/preprocess.py
• python ./scripts/preprocess.py

Results

loss = top1, rnn-size = 100, epochs = 3

• Recall@20: 0.4243443414791227
• MRR@20: 0.10790868224055285

References

• Main source: https://github.com/Songweiping/GRU4Rec_TensorFlow
• The original implementation: https://github.com/hidasib/GRU4Rec