DocDataset

class lightning_ir.data.dataset.DocDataset(doc_dataset: str, num_docs: int | None = None)

Bases: IRDataset, DataParallelIterableDataset

__init__(doc_dataset: str, num_docs: int | None = None) → None

Methods

__init__(doc_dataset[, num_docs])

Attributes

DASHED_DATASET_MAP
dataset_id
docs
docs_dataset_id
qrels
queries