TupleDataset

class lightning_ir.data.dataset.TupleDataset(tuples_dataset: str, targets: Literal['order', 'score'] = 'order', num_docs: int | None = None)

Bases: IRDataset, IterableDataset

__init__(tuples_dataset: str, targets: Literal['order', 'score'] = 'order', num_docs: int | None = None) → None

Methods

__init__(tuples_dataset[, targets, num_docs])
parse_sample(sample)

Attributes

DASHED_DATASET_MAP
dataset_id
docs
docs_dataset_id
qrels
queries