TorchSparseSearchConfig

class lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseSearchConfig(k: int = 10)

Bases: ExactSearchConfig

Configuration for the Torch-based sparse searcher.

__init__(k: int = 10) → None

Initialize the SearchConfig.

Parameters:

k (int) – The number of top documents to retrieve. Defaults to 10.

Methods

Attributes

SUPPORTED_MODELS

search_class

alias of TorchSparseSearcher