GRU4REC-pytorch

GRU4REC-pytorch copied to clipboard

Update evaluation.py

3

the mrr metric cannot work with np.mean as mrr is a tensor value therefore I Have proposed this change.

MasterKinjalk

Question about the difference of the GRU4Rec and GRU4Rec+

1

Hi,I have a question about how to run gru4rec and gru4rec+ with your code.From the paper,we can know these two methods vary in Loss Function and Sampling Strategy while the...

zyh981022

Hi, i have read your bench mark file https://docs.google.com/spreadsheets/d/19z6zFEY6pC0msi3wOQLk_kJsvqF8xnGOJPUGhQ36-wI/edit#gid=1577545099, and may i ask you if this code is GRU4Rec or GRU4Rec+, and can the Improved GRU4Rec bench marks be achieved...

H13U1999

Evaluation does not reset hidden states.

1

``` with torch.no_grad(): hidden = self.model.init_hidden() for ii, (input, target, mask) in tqdm(enumerate(dataloader), total=len(dataloader.dataset.df) // dataloader.batch_size, miniters = 1000): #for input, target, mask in dataloader: input = input.to(self.device) target =...

lyf-1

question about mask in data_loader

3

Hi, I have a question about the "mask" in DataLoader(), ``` for i in range(minlen - 1): idx_input = idx_target idx_target = df.item_idx.values[start + i + 1] input = torch.LongTensor(idx_input)...

yuqianghan