A problem in training
Opened this issue · 1 comments
fatemeakbari commented
Hi
in forward function you first get users and then movies:
def forward(self, users, movies, minmax=None):
but in training you first get column 1 which is movies id and then column 0 which is user id:
outputs = net(x_batch[:, 1], x_batch[:, 0], minmax)
I wanted to know if there is a reason behind this or it is a problem.
thank you very much for your time
ahmedbr commented
outputs = net(x_batch[:, 1], x_batch[:, 0], minmax)
Just switch between 0 & 1. And everything will be fine.