TorchDenseSearchConfig

class lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseSearchConfig(k: int = 10)

Bases: ExactSearchConfig

Configuration for the TorchDenseSearcher.

__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 TorchDenseSearcher