This repository contains the data and the notebooks of an implementation of a model described in [1]. The blog post is here.

The data is the movielens 2k dataset by - GroupLens research group.

1. Merging.ipynb contains the code for the merging of the data.

2. Preprocessing.ipynb conatins the code for some basic preprocessing.

3. Modeling.ipynb is the modeling part takes place.

4. Results.ipynb contains the visualizations of the results.

[1] P. Gopalan, J. Hofman, D. Blei. Sacalable Recommendation with Hierarchical Poisson Factorization. Proceedings of the Thirty-First Conference on Uncertainty in Artificial Intelligence. 2015