FaissIVFIndexer

class lightning_ir.retrieve.faiss_indexer.FaissIVFIndexer(index_dir: Path, index_config: FaissIVFIndexConfig, bi_encoder_config: BiEncoderConfig, verbose: bool = False)

Bases: FaissIndexer

__init__(index_dir: Path, index_config: FaissIVFIndexConfig, bi_encoder_config: BiEncoderConfig, verbose: bool = False) → None

Methods

__init__(index_dir, index_config, ...[, verbose])
add(index_batch, output)
process_embeddings(embeddings)
save()
set_verbosity([verbose])
to_cpu()
to_gpu()

Attributes

INDEX_FACTORY