ShopRunner/collie
A library for preparing, training, and evaluating scalable deep learning hybrid recommender systems using PyTorch.
PythonBSD-3-Clause
Issues
- 4
Hybrid model item_metadata with nulls
#47 opened by ahuds001 - 1
Test notebooks run as part of CI checks
#18 opened by nathancooperjones - 10
- 1
[Question] Can I use Collie to get shopping cart complementary products on an e-shop?
#38 opened by almirb - 11
Including user metadata in hybrid model
#21 opened by lgpreston75 - 0
Multi-GPU model training
#46 opened by nathancooperjones - 4
Error when using stratified_split[BUG]
#41 opened by sgaseretto - 5
Question regarding production inference
#42 opened by mhashas - 0
Adaptive negative sampling!
#27 opened by nathancooperjones - 0
Add architecture that allows embeddings to be stored on the CPU but training done on the GPU
#26 opened by nathancooperjones - 0
Add support for explicit data
#10 opened by nathancooperjones - 1