searcher

Base searcher class and configuration for retrieval tasks.

Functions

cat_arange(arange_starts, arange_ends)

Concatenates arange tensors into a single tensor.

Classes

ApproximateSearchConfig([k, candidate_k, ...])

Configuration class for approximate searchers in the Lightning IR framework.

ApproximateSearcher(index_dir, ...[, use_gpu])

ExactSearchConfig([k])

Configuration class for exact searchers in the Lightning IR framework.

ExactSearcher(index_dir, search_config, module)

Searcher that retrieves documents using exact matching of query embeddings.

SearchConfig([k])

Configuration class for searchers in the Lightning IR framework.

Searcher(index_dir, search_config, module[, ...])

Base class for searchers in the Lightning IR framework.