Collaborative Filtering, Matrix Factorization, Recommender Systems, Gradient Descent, Neural Networks.
torch.nn.Linear(num_items, K) torch.nn.init.xavier_uniform_(self.V_layer.weight) # Feed-forward hidden layers self.fc1 = torch.nn.Linear(2 * K, K) # Concat U, V to arbitrary K self.fc2 = torch.nn.Linear(K, 1) # To 1 as needs scalar value self.MSE = torch.nn.MSELoss() # Params are the embedding