PlaidSearchConfig

class lightning_ir.retrieve.plaid.plaid_searcher.PlaidSearchConfig(k: int, candidate_k: int = 256, n_cells: int = 1, centroid_score_threshold: float = 0.5)

Bases: SearchConfig

Configuration class for Plaid searchers in the Lightning IR framework.

__init__(k: int, candidate_k: int = 256, n_cells: int = 1, centroid_score_threshold: float = 0.5) → None

Initialize the PlaidSearchConfig.

Parameters:

• k (int) – Number of top documents to retrieve.

• candidate_k (int) – Number of candidate documents to consider for scoring. Defaults to 256.

• n_cells (int) – Number of cells to use for centroid retrieval. Defaults to 1.

• centroid_score_threshold (float) – Threshold for filtering candidates based on centroid scores. Defaults to 0.5.

Methods

__init__(k[, candidate_k, n_cells, ...])

Initialize the PlaidSearchConfig.

Attributes

SUPPORTED_MODELS

search_class

alias of PlaidSearcher