recbole.data.dataloader.abstract_dataloader

class recbole.data.dataloader.abstract_dataloader.AbstractDataLoader(config, dataset, sampler, shuffle=False)

Bases: object

AbstractDataLoader is an abstract object which would return a batch of data which is loaded by Interaction when it is iterated. And it is also the ancestor of all other dataloader.

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.

dataset

The dataset of this dataloader.

Type

Dataset

shuffle

If True, dataloader will shuffle before every epoch.

Type

bool

pr

Pointer of dataloader.

Type

int

step

The increment of pr for each batch.

Type

int

batch_size

The max interaction number for all batch.

Type

int

property pr_end

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

set_batch_size(batch_size)

Reset the batch_size of the dataloader, but it can’t be called when dataloader is being iterated.

Parameters

batch_size (int) – the new batch_size of dataloader.

update_config(config)

Update configure of dataloader, such as batch_size, step etc.

Parameters

config (Config) – The new config of dataloader.

class recbole.data.dataloader.abstract_dataloader.NegSampleDataLoader(config, dataset, sampler, shuffle=True)

Bases: recbole.data.dataloader.abstract_dataloader.AbstractDataLoader

NegSampleDataLoader is an abstract class which can sample negative examples by ratio. It has two neg-sampling method, the one is 1-by-1 neg-sampling (pair wise), and the other is 1-by-multi neg-sampling (point wise).

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.

get_model(model)