def predict(self, interaction):
    user = interaction[self.USER_ID]
    item = interaction[self.ITEM_ID]
    (user_e, item_e) = self.forward(user, item)
    return torch.mul(user_e, item_e).sum(dim=1)
def predict(self, interaction):
    user = interaction[self.USER_ID]
    item = interaction[self.ITEM_ID]
    (user_e, item_e) = self.forward(user, item)
    return torch.mul(user_e, item_e).sum(dim=1)