An open API service indexing awesome lists of open source software.

https://github.com/fuhailin/probabilistic-matrix-factorization

Python Implementation of Probabilistic Matrix Factorization(PMF) Algorithm for building a recommendation system using MovieLens ml-100k | GroupLens dataset
https://github.com/fuhailin/probabilistic-matrix-factorization

matrix-factorization recommendation-system

Last synced: about 2 months ago
JSON representation

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

Awesome Lists containing this project

README

          

## 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](https://github.com/fuhailin/Probabilistic-Matrix-Factorization/blob/master/docs/PMF.png)

The dataset is a copy of the MovieLens | GroupLens
dataset in the `MovieLens 100k | GroupLens `

### 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.