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)

Bases: object

A batch of search data that combines multiple QuerySample instances. Optionaly includes document ids and qrels.

query_ids

list of query ids.

Type:

Sequence[str]

queries

list of query texts.

Type:

Sequence[str]

doc_ids

Optional list of list of document ids. Defaults to None.

Type:

Sequence[Sequence[str]] | None

qrels

Optional list of dictionaries mapping document ids to relevance labels. Defaults to None.

Type:

list[dict[str, int]] | None

__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