Fix pickling GPU models

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 pickling GPU models #632

Closed benfred closed 1 year ago

benfred commented 1 year ago

GPU models weren't default initializing members like _knn appropriately when being loaded from pickle appropriately. Fix and update the unittest to test this out.