Skip to content

Latest commit

 

History

History

02_model_collaborative_filtering

Deep dive in collaborative filtering algorithms

In this directory, notebooks are provided to give a deep dive of collaborative filtering recommendation algorithms. The notebooks make use of the utility functions (recommenders) available in the repo.

Notebook Environment Description
als_deep_dive PySpark Deep dive on the ALS algorithm and implementation.
baseline_deep_dive --- Deep dive on baseline performance estimation.
cornac_bivae_deep_dive Python CPU, GPU Deep dive on the BiVAE algorithm and implementation.
cornac_bpr_deep_dive Python CPU Deep dive on the BPR algorithm and implementation.
fm_deep_dive Python CPU Deep dive into factorization machine (FM) and field-aware FM (FFM) algorithm.
lightfm_deep_dive Python CPU Deep dive into matrix factorization model with LightFM.
lightgcn_deep_dive Python CPU, GPU Deep dive on a LightGCN algorithm and implementation.
multi_vae_deep_dive Python CPU, GPU Deep dive on the Multinomial VAE algorithm and implementation.
ncf_deep_dive Python CPU, GPU Deep dive on a NCF algorithm and implementation.
rbm_deep_dive Python CPU, GPU Deep dive on the rbm algorithm and its implementation.
sar_deep_dive Python CPU Deep dive on the SAR algorithm and implementation.
standard_vae_deep_dive Python CPU, GPU Deep dive on the Standard VAE algorithm and implementation.
surprise_svd_deep_dive Python CPU Deep dive on a SVD algorithm and implementation.

Details on model training are best found inside each notebook.