SearchBatch

class lightning_ir.data.data.SearchBatch(query_ids: Sequence[str], queries: Sequence[str], doc_ids: Sequence[Sequence[str]] | None = None, qrels: List[Dict[str, int]] | None = None)[source]

Bases: object

__init__(query_ids: Sequence[str], queries: Sequence[str], doc_ids: Sequence[Sequence[str]] | None = None, qrels: List[Dict[str, int]] | None = None) None

Methods

__init__(query_ids, queries[, doc_ids, qrels])

Attributes

doc_ids

qrels

query_ids

queries