Complete the Look: Scene-based Complementary Product Recommendation
- PyTorch implementation for Complete the Look: Scene-based Complementary Product Recommendation
- It mainly depends on the paper description, not an official implementation
- Reference. Wang-Cheng Kang et. al., 2019, Complete the Look: Scene-based Complementary Product Recommendation
Dependencies
- Python == 3.6.9
- PyTorch == 1.9.1
- CUDA == 11.x
Command-line Interface
Train
Inference
Code Format
- black == 21.9b0
References
- Wang-Cheng Kang et. al., 2019, Complete the Look: Scene-based Complementary Product Recommendation
- STL(Shop the Look) Dataset
TODO
- Train codes
- Upload evaluation result
- Faiss retrieval phase
- Notes for different implementation points (TFLite, ...)
- Data preprocessing pipeline