The hybrid model combining stacked denoising autoencoder with matrix factorization is applied, to predict the customer purchase behavior in the future month according to the purchase history and user information in the Santander dataset.
Switch branches/tags
Nothing to show
Clone or download
Latest commit dd332e3 Aug 4, 2018

README.md

Collaborative-Deep-Learning-for-Recommender-Systems

The hybrid model combining stacked denoising autoencoder (SDAE) with matrix factorization (MF) is applied, to predict the customer purchase behavior in the future month according to the purchase history and user information in the Santander dataset. A blog post for some follow up discussions can be found here.

This work is contributed by Sampath Chanda, Suyin Wang and Xiaoou Zhang.

The user information is generated in "matrix_factorization.ipynb", and the rating marix is generated in "rating_matrix.py". The main code for the SDAE-MF hybrid model can be found in "mf_auto_mono.py" for one-hidden-layer SDAE and "mf_auto.py" for three-hidden-layer SDAE.

Update: in "mf_auto_mono_v2.py" and "mf_auto_v2.py", imporved the speed of matrix factorization by taking advantage of the sparsity of the rating matrix.