Neural-Collaborative-Filtering

Neural-Collaborative-Filtering copied to clipboard

a question about results

1

Hello! I have a question to ask. I set the number of epochs to 200, and I initially thought that the overall trend of HR and NDCG would continuously increase...

zkLyons

problem

1

class NeuMF(nn.Module): def __init__(self, args, num_users, num_items): ... def init_weight(self): ... def forward(self, user_indices, item_indices): user_embedding_mlp = self.embedding_user_mlp(user_indices) item_embedding_mlp = self.embedding_item_mlp(item_indices) user_embedding_mf = self.embedding_user_mf(user_indices) item_embedding_mf = self.embedding_item_mf(item_indices) mlp_vector = torch.cat([user_embedding_mlp,...

zkLyons