pytorch_playground icon indicating copy to clipboard operation
pytorch_playground copied to clipboard

A problem in training

Open fatemeakbari opened this issue 4 years ago • 1 comments

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

fatemeakbari avatar Aug 10 '19 10:08 fatemeakbari

outputs = net(x_batch[:, 1], x_batch[:, 0], minmax) Just switch between 0 & 1. And everything will be fine.

ahmedbr avatar Dec 18 '19 12:12 ahmedbr