A set of Jupyter notebooks demonstrating a simple Keras implmentation of matrix factorization for collaborative filtering. The notebooks use the MovieLens 1M Dataset [1] to show the effectiveness of the architecture. Using 120-dimensional embeddings for users and movies, we achieve an RMSE of 0.862 on a held-out validation set after 15 epochs, taking under 18 minutes on an AWS EC2 g2.2xlarge instance with an NVIDIA GRID K520 GPU.
The notebooks provide the following workflows:
- MovieLens 1M ETL: loads and processes user, movie and ratings data to prepare them for input into the Keras model.
- MovieLens Training: trains an instance of CFModel using the prepared MovieLens data.
- MovieLens Recommendations: shows recommendations generated using the trained model for a given test user.
- Python 2.7
- A copy of the MovieLens 1M dataset, downloaded from [2].
- pandas (0.18.1)
- matplotlib (1.5.2)
- keras (1.0.5)
- numpy (1.11.1)
- h5py (2.6.0)
MIT. See the LICENSE file for the copyright notice.
[1] F. Maxwell Harper and Joseph A. Konstan. 2015. The MovieLens Datasets: History and Context. ACM Trans. Interact. Intell. Syst. 5, 4, Article 19 (December 2015).
[2] MovieLens 1M Dataset. https://grouplens.org/datasets/movielens/1m/. Last downloaded 2016-08-14.