Skip to content

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)[source]

Bases: object

__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`