benfred / implicit

Fast Python Collaborative Filtering for Implicit Feedback Datasets
https://benfred.github.io/implicit/
MIT License
3.57k stars 612 forks source link

Fix `recalculate_item` / `recalculate_user` on the GPU #598

Closed benfred closed 2 years ago

benfred commented 2 years ago

The recalculate_user or recalculate_item functionality didn't work on the GPU AlternatingLeastSquares model, if the model was created from a saved version of converted from a CPU model. Fix and add a unittest that would have caught this

Fixes #597