Based on Olivier Moindrot blog post
Dataset: Stanford Online Products
How to use:
For more convenient experimenting use Docker and docker-compose to start scripts.
- Tweak config depending on the task to be solved
- Train embedding model using
train_model.py - Evaluate model using appropriate k in
evaluate.py