Skip to main content

Module cross_encoder

Module cross_encoder 

Source
Expand description

Cross-encoder reranking implementation Cross-Encoder reranking for improved retrieval accuracy

Cross-encoders jointly encode query and document, providing more accurate relevance scores than bi-encoder approaches. This implementation provides a trait-based interface that can be backed by ONNX models, API calls, or other implementations.

Reference: “Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks” Reimers & Gurevych (2019)

Structs§

ConfidenceCrossEncoder
Confidence-based cross-encoder implementation
CrossEncoderConfig
Configuration for cross-encoder reranking
RankedResult
Result of cross-encoder reranking with confidence score
RerankingStats
Statistics about reranking performance

Traits§

CrossEncoder
Cross-encoder trait for reranking retrieved results