/pyPMF

Implementing of probabilistic matrix factorization via python

Primary LanguagePython

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