Home

Awesome

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.