FaissIVFIndexConfig

class lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFIndexConfig(num_train_embeddings: int | None = None, num_centroids: int = 262144, ef_construction: int = 40)

Bases: _FaissTrainIndexConfig

__init__(num_train_embeddings: int | None = None, num_centroids: int = 262144, ef_construction: int = 40) → None

Methods

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