Searcher

class lightning_ir.retrieve.searcher.Searcher(index_dir: Path | str, search_config: SearchConfig, module: BiEncoderModule, use_gpu: bool = True)

Bases: ABC

__init__(index_dir: Path | str, search_config: SearchConfig, module: BiEncoderModule, use_gpu: bool = True) → None

Methods

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

Attributes

num_embeddings