RankBatch

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

Bases: object

A batch of ranking data combining multiple RankSample instances

queries

List of query texts.

Type:

Sequence[str]

docs

List of list of document texts.

Type:

Sequence[Sequence[str]]

query_ids

Optional list of query ids. Defaults to None.

Type:

Sequence[str] | None

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__(queries: Sequence[str], docs: Sequence[Sequence[str]], query_ids: Sequence[str] | None = None, doc_ids: Sequence[Sequence[str]] | None = None, qrels: List[Dict[str, int]] | None = None) → None

Methods

__init__(queries, docs[, query_ids, ...])

Attributes

doc_ids
qrels
query_ids
queries
docs