Visually Explainable Recommendations
Python 2.7
Pytorch
The data is taken from here. We have specifically worked with Clothing, Shoes and Jewelry data.
There are some files that don't have explanation as they are not directly involved with the project
- extract_imageurls.py - This script extracts imageurls into a file from the metadata available
- collect_images.py - This script downloads all those images from those urls into a folder
- resize_images.py - This script resizes them all to be of the same size
- user_item_pairs.py - This script extracts the user, items interaction from the metadata
- model.py - This file mainly consists of the model and helper funtions like training function
- constants.py - This file consists of environment variables
- main.py - This file is the one that is to be run to learn the model