MRMA-tensorflow

Overview

• The tensorflow implementation of MRMA (Mixture-Rank Matrix Approximation for Collaborative Filtering) from Mixture-Rank Matrix Approximation for Collaborative Filtering.
• If you have curious about this paper, refer Unofficial Slide.
• I tested MovieLens 100k and 1M dataset.

Environment

• I've tested this code on Python3.5 (tensorflow 1.12) and Ubuntu 16.04.

Run

• pretrain:

python3 -m pmf.trainer

• train:

python3 -m mrma.trainer