Collaborative Filtering, Matrix Factorization, Recommender Systems, Gradient Descent, Neural Networks.
above more simply with Pytorch (will achieve the same performance): # Model definition class PytorchLinearModel(torch.nn.Module): def __init__(self, num_users, num_items, K, lr): super().__init__() # Set U and V as parameters of this model self.U = torch.nn.Parameter(torch.zeros((num_users, K)))