IndexBatch

class lightning_ir.data.data.IndexBatch(doc_ids: Sequence[str], docs: Sequence[str])[source]

Bases: object

A batch of index that combines multiple DocSample instances

Parameters:

  • doc_ids (Sequence[str]) – List of document ids

  • docs (Sequence[str]) – List of document texts

__init__(doc_ids: Sequence[str], docs: Sequence[str]) → None

Methods

__init__(doc_ids, docs)

Attributes

doc_ids

docs