FaissIVFPQIndexConfig

class lightning_ir.retrieve.faiss_indexer.FaissIVFPQIndexConfig(num_train_embeddings: int | None = None, num_centroids: int = 262144, ef_construction: int = 40, num_subquantizers: int = 16, n_bits: int = 8)

Bases: FaissIVFIndexConfig

__init__(num_train_embeddings: int | None = None, num_centroids: int = 262144, ef_construction: int = 40, num_subquantizers: int = 16, n_bits: int = 8) → None

Methods

__init__([num_train_embeddings, ...])
from_pretrained(index_dir)
save(index_dir)
to_dict()