Collaborative Filtering, Matrix Factorization, Recommender Systems, Gradient Descent, Neural Networks.
PytorchNonLinearModel(torch.nn.Module): def __init__(self, num_users, num_items, K, lr): super().__init__() # Embedding layers (same conceptual purpose from the linear model) self.U_layer = torch.nn.Linear(num_users, K) torch.nn.init.xavier_uniform_(self.U_layer.weight) self.V_layer =