Collaborative Filtering, Matrix Factorization, Recommender Systems, Gradient Descent, Neural Networks.
layers (U, V) and other layers (fc1, fc2) self.optimizer = torch.optim.SGD(self.parameters(), lr=lr) def forward(self, U_one_hots, V_one_hots): # Embedding layers U_latent = self.U_layer(U_one_hots) # Indexes the U embedding layer V_latent = self.V_layer(V_one_hots) # Indexes the V embedding layer