Hi, I always wondered how ecommerce websites built their recommendation engines that recommend similar models to us. So, I took it as a challenge to build myself a small model.
Here’s a small static deployed model for product recommendations, ResNet50 was used and the transfer learning technique is used to train the model.
The deployed model is just a small prototype but I liked how accurate it was! The similarity scores of the products are also listed.
Here is the working prototype on heroku : Product-recommendations
For more details and the model with more than 2000+ images used for training: Click here
1 post - 1 participant