• Stars
    star
    165
  • Rank 227,584 (Top 5 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created over 7 years ago
  • Updated over 5 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

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

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.