Struct BiEncoder 

pub struct BiEncoder<F> { /* private fields */ }

Bi-Encoder for efficient dense retrieval.

Encodes queries and documents separately, allowing pre-computation of document embeddings for fast retrieval.

Architecture

Query  ─┬─> Encoder ─> Query Embedding  ─┐
        │                                 ├─> Similarity Score
Document ─> Encoder ─> Doc Embedding   ─┘

Implementations

impl<F> BiEncoder<F>

where F: Fn(&[f32]) -> Vec<f32>,

pub fn new(encode_fn: F, similarity: SimilarityMetric) -> Self

Create bi-encoder with custom encoding function.

pub fn encode(&self, input: &[f32]) -> Vec<f32>

Encode a single input.

pub fn encode_batch(&self, inputs: &[Vec<f32>]) -> Vec<Vec<f32>>

Encode a batch of inputs.

pub fn similarity(&self, a: &[f32], b: &[f32]) -> f32

Compute similarity between two embeddings.

pub fn retrieve<T: Clone>( &self, query: &[f32], corpus: &[(T, Vec<f32>)], top_k: usize, ) -> Vec<(T, f32)>

Retrieve top-k most similar documents.

Trait Implementations

impl<F: Debug> Debug for BiEncoder<F>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.