Skip to content
GRu4Rec in TensorFlow
Branch: master
Clone or download
Pull request Compare This branch is 1 commit ahead of codejitsu:master.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
scripts
.gitignore
README.md
Session4Rec.ipynb
defaults.py
evaluation.py
model_session4rec.py
model_simple_rnn.py
requirements.txt
run_eval.sh
run_train.sh

README.md

session4rec

GRu4Rec in Tensorflow

Get data

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

You can’t perform that action at this time.