SparseIndex

class lightning_ir.retrieve.sparse.sparse_searcher.SparseIndex(index_dir: Path, similarity_function: 'dot' | 'cosine', use_gpu: bool = False)

Bases: object

__init__(index_dir: Path, similarity_function: 'dot' | 'cosine', use_gpu: bool = False) → None

Methods

__init__(index_dir, similarity_function[, ...])
cosine_similarity(x, y)
dot_similarity(x, y)
score(embeddings)
to_gpu()

Attributes

num_embeddings