PlaidSearcher

class lightning_ir.retrieve.plaid.plaid_searcher.PlaidSearcher(index_dir: Path | str, search_config: PlaidSearchConfig, module: BiEncoderModule, use_gpu: bool = False)

Bases: Searcher

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

Methods

__init__(index_dir, search_config, module[, ...])
candidate_retrieval(query_embeddings)
filter_candidates(centroid_scores, doc_idcs, ...)
search(output)
to_gpu()

Attributes

num_embeddings