/Probabilistic-Matrix-Factorization

Python Implementation of Probabilistic Matrix Factorization(PMF) Algorithm for building a recommendation system using MovieLens ml-100k | GroupLens dataset

Primary LanguagePythonApache License 2.0Apache-2.0

Python Implementation of Probabilistic Matrix Factorization Algorithm

The code attempts to implement the following paper:

Mnih, A., & Salakhutdinov, R. (2007). Probabilistic matrix factorization. In Advances in neural information processing systems (pp. 1257-1264).

Probabilistic Matrix Factorization in Python with MovieLens dataset

image

The dataset is a copy of the MovieLens | GroupLens dataset in the MovieLens 100k | GroupLens <http://files.grouplens.org/datasets/movielens/ml-100k.zip/>

Reference:

  1. Mnih, A., & Salakhutdinov, R. (2007). Probabilistic matrix factorization. In Advances in neural information processing systems (pp. 1257-1264).
  2. Salakhutdinov, R. Probabilistic matrix factorization in Matlab. http://www.utstat.toronto.edu/~rsalakhu/BPMF.html.