FaissSearcher

class lightning_ir.retrieve.faiss_searcher.FaissSearcher(index_dir: Path | str, search_config: FaissSearchConfig, module: BiEncoderModule, use_gpu: bool = False)

Bases: Searcher

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

Methods

__init__(index_dir, search_config, module[, ...])
candidate_retrieval(query_embeddings)
gather_imputation(candidate_doc_idcs, ...)
intra_ranking_imputation(candidate_scores, ...)
search(output)
to_gpu()

Attributes

doc_is_single_vector
num_embeddings