Struct ProbabilisticRouter 

pub struct ProbabilisticRouter { /* private fields */ }

Probabilistic routing layer for HNSW search.

Implementations

impl ProbabilisticRouter

pub fn new(config: ProbabilisticRoutingConfig) -> Self

Create a new probabilistic router.

pub fn filter_neighbors<'a>( &mut self, query: &[f32], current_pos: &[f32], current_best_dist: f32, neighbors: &'a [(u32, &[f32])], ) -> Vec<(u32, &'a [f32], f32)>

Filter neighbors to test based on probability.

Returns indices of neighbors that should be tested.

pub fn search( &mut self, query: &[f32], entry_point: u32, entry_pos: &[f32], get_neighbors: impl Fn(u32) -> Vec<(u32, Vec<f32>)>, ef: usize, ) -> Vec<(u32, f32)>

Perform probabilistic beam search.

This is a search implementation that uses probabilistic routing internally.

pub fn stats(&self) -> &ProbabilisticStats

Get statistics.

pub fn reset_stats(&mut self)

Reset statistics.

Trait Implementations

impl Debug for ProbabilisticRouter

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

Formats the value using the given formatter.