PlaidIndexConfig

class lightning_ir.retrieve.plaid.plaid_indexer.PlaidIndexConfig(num_centroids: int, num_train_embeddings: int, k_means_iters: int = 4, n_bits: int = 2, seed: int = 42)

Bases: IndexConfig

__init__(num_centroids: int, num_train_embeddings: int, k_means_iters: int = 4, n_bits: int = 2, seed: int = 42) → None

Methods

__init__(num_centroids, num_train_embeddings)
from_pretrained(index_dir)
save(index_dir)
to_dict()