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