fde2f66676f960782c993f7148927c4a4197ab10,spotlight/factorization/explicit.py,ExplicitFactorizationModel,fit,#ExplicitFactorizationModel#,85

Before Change

        if self._optimizer is None:
            self._optimizer = optim.Adam(self._net.parameters())

        if self._loss == "pointwise":
            loss_fnc = pointwise_loss
        elif self._loss == "bpr":
            loss_fnc = bpr_loss
        elif self._loss == "hinge":
            loss_fnc = hinge_loss

        for epoch_num in range(self._n_iter):

            users, items, ratings = shuffle(*(interactions.row,
                                              interactions.col,

After Change

        
        

        user_ids = interactions.user_ids.astype(np.int64)
        item_ids = interactions.item_ids.astype(np.int64)

        (self._num_users,
         self._num_items) = (interactions.num_users,

Instances