SparseSearcher

class lightning_ir.retrieve.sparse.sparse_searcher.SparseSearcher(index_dir: Path, search_config: SparseSearchConfig, module: BiEncoderModule, use_gpu: bool = True)[source]

Bases: Searcher

__init__(index_dir: Path, search_config: SparseSearchConfig, module: BiEncoderModule, use_gpu: bool = True) → None[source]

Methods

`__init__`(index_dir, search_config, module[, ...])
`search`(output)
`to_gpu`()

Attributes

`doc_is_single_vector`
`num_embeddings`