PlaidIndexer

class lightning_ir.retrieve.plaid.plaid_indexer.PlaidIndexer(index_dir: Path, index_config: PlaidIndexConfig, module: BiEncoderModule, verbose: bool = False)

Bases: Indexer

__init__(index_dir: Path, index_config: PlaidIndexConfig, module: BiEncoderModule, verbose: bool = False) → None

Methods

__init__(index_dir, index_config, module[, ...])
add(index_batch, output)
process_embeddings(embeddings)
save()