Skip to content

Latest commit

 

History

History

product-recommender

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 

Product Recommender using Collaborative Filtering and LanceDB

Use LanceDB and collaborative filtering to recommend products based on a user's past buying history. We used the Instacart dataset as our data for this example. Colab walkthrough - Open In Colab

To run this example, you must first create a Kaggle account. Then, go to the 'Account' tab of your user profile and select 'Create New Token'. This will trigger the download of kaggle.json, a file containing your API credentials.

Add Kaggle credentials to ~/.kaggle/kaggle.json on Linux, OSX, and other UNIX-based operating systems or C:\Users\<Windows-username>\.kaggle\kaggle.json for Window's users.

Python

Download the dataset (you must have requirements installed first!) You will need to accept the rules of the instacart-market-basket-analysis competition, which you can do so here.

kaggle competitions download -c instacart-market-basket-analysis

Run the script

python main.py
Argument Default Value Description
factors 100 dimension of latent factor vectors
regularization 0.05 strength of penalty term
iterations 50 number of iterations to update
num-threads 1 amount of parallelization
num-partitions 256 number of partitions of the index
num-sub-vectors 16 number of sub-vectors (M) that will be created during Product Quantization (PQ)