data.py 文件源码

def test_to_coo(self, tensor_mode=False):
        userid, itemid, feedback = self.fields
        test_data = self.test.testset

        user_idx = test_data[userid].values.astype(np.intp)
        item_idx = test_data[itemid].values.astype(np.intp)
        fdbk_val = test_data[feedback].values

        if tensor_mode:
            fdbk_idx = self.index.feedback.set_index('old').loc[fdbk_val, 'new'].values
            if np.isnan(fdbk_idx).any():
                raise NotImplementedError('Not all values of feedback are present in training data')
            else:
                fdbk_idx = fdbk_idx.astype(np.intp)
            test_coo = (user_idx, item_idx, fdbk_idx)
        else:
            test_coo = (user_idx, item_idx, fdbk_val)

        return test_coo