Collaborative Filtering, Matrix Factorization, Recommender Systems, Gradient Descent, Neural Networks.
R_hat = MF_model.forward(U_one_hots, V_one_hots) # Calc MSE loss of this reconstruction train_loss = MF_model.calculate_loss(R, R_hat) # Calc grad and update MF_model.optimize(train_loss) Sample Performance on the Movie Lens Dataset Here is a performance sample from the MovieLens 100K dataset.