pyPMF

• Implementation of probabilistic matrix factorization via python.
• This version fits in MovieLens data format and the dataset is attached as well.
• Note the model will save itself as pickle file every epoch and print training RMSE every 50 iterations.

Reference:

1. Mnih, A., & Salakhutdinov, R. (2007). Probabilistic matrix factorization. In Advances in neural information processing systems (pp. 1257-1264).
2. F. Maxwell Harper and Joseph A. Konstan. 2015. The MovieLens Datasets: History and Context. ACM Transactions on Interactive Intelligent Systems (TiiS) 5, 4, Article 19 (December 2015), 19 pages. DOI=http://dx.doi.org/10.1145/2827872