recbole.data.dataloader.user_dataloader

class recbole.data.dataloader.user_dataloader.UserDataLoader(config, dataset, sampler, shuffle=False)[source]

Bases: recbole.data.dataloader.abstract_dataloader.AbstractDataLoader

UserDataLoader will return a batch of data which only contains user-id when it is iterated.

Parameters

  • config (Config) – The config of dataloader.

  • dataset (Dataset) – The dataset of dataloader.

  • sampler (Sampler) – The sampler of dataloader.

  • shuffle (bool, optional) – Whether the dataloader will be shuffle after a round. Defaults to False.

shuffle

Whether the dataloader will be shuffle after a round. However, in UserDataLoader, it’s guaranteed to be True.

Type

bool

property pr_end

This property marks the end of dataloader.pr which is used in __next__().