Fit function in gru4rec.py missing data sort

hidasib / GRU4Rec

GRU4Rec is the original Theano implementation of the algorithm in "Session-based Recommendations with Recurrent Neural Networks" paper, published at ICLR 2016 and its follow-up "Recurrent Neural Networks with Top-k Gains for Session-based Recommendations". The code is optimized for execution on the GPU.

Other

747 stars 222 forks source link

Fit function in gru4rec.py missing data sort #35

Closed dawn-farsi closed 5 years ago

dawn-farsi commented 5 years ago

Thank you for sharing your great job. In gru4rec.py file, I think data should be sorted by session_key and item_key after line 508 to be aligned to offset_sessions.

TauraiUCB commented 2 years ago

Hi @sfarsinezhad I have encountered the same problem during testing.

 start = offset_sessions[iters]
IndexError: index 2 is out of bounds for axis 0 with size 2

How precisely were you able to fix it?