Skip to content

Latest commit

 

History

History
17 lines (14 loc) · 1.35 KB

README.md

File metadata and controls

17 lines (14 loc) · 1.35 KB

Model

In this directory, notebooks are provided to give a deep dive into training models using different algorithms such as Alternating Least Squares (ALS) and Singular Value Decomposition (SVD) using Surprise python package. The notebooks make use of the utility functions (reco_utils) 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.
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.
surprise_svd_deep_dive Python CPU Deep dive on a SVD algorithm and implementation.
vowpal_wabbit_deep_dive Python CPU Deep dive into using Vowpal Wabbit for regression and matrix factorization.

Details on model training are best found inside each notebook.