transformer_rankers.datasets.dataset.QueryPosDocNegDocDataLoader

class transformer_rankers.datasets.dataset.QueryPosDocNegDocDataLoader(train_df, val_df, test_df, tokenizer, negative_sampler_train, negative_sampler_val, task_type, train_batch_size, val_batch_size, max_seq_len, sample_data, cache_path)

Bases: transformer_rankers.datasets.dataset.AbstractDataloader

__init__(train_df, val_df, test_df, tokenizer, negative_sampler_train, negative_sampler_val, task_type, train_batch_size, val_batch_size, max_seq_len, sample_data, cache_path)

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(train_df, val_df, test_df, …)	Initialize self.
get_pytorch_dataloaders()