This is the source code for machine learning course final project. We built a recommender system based on the data set in kaggle competition about H&M. We mainly applied content based filtering method and compared with popularity method plus ranking with light gradient boosting. In content based filtering we used text data, image data and tabular data to find similar products with different embedding such as vgg16 for image and bert, TFIDF for text.
-
Notifications
You must be signed in to change notification settings - Fork 0
plingyy/Recommender_System
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published