QueryDataset

class lightning_ir.data.dataset.QueryDataset(query_dataset: str, num_queries: int | None = None)[source]

Bases: IRDataset, DataParallelIterableDataset

__init__(query_dataset: str, num_queries: int | None = None) → None[source]

Methods

__init__(query_dataset[, num_queries])

Attributes

DASHED_DATASET_MAP

dataset_id

docs

docs_dataset_id

qrels

queries