Struct PQIndex 

pub struct PQIndex { /* private fields */ }

PQ-based index for compressed vector search

Implementations

impl PQIndex

pub fn new(config: PQConfig) -> PQIndex

Create a new PQ index

pub fn with_originals(self) -> PQIndex

Create and enable original vector storage for reranking

pub fn train(&mut self, vectors: &[Vec<f32>])

Train the index

pub fn add(&mut self, vector: Vec<f32>) -> u64

Add a vector

pub fn add_with_id(&mut self, id: u64, vector: Vec<f32>)

Add a vector with ID

pub fn add_batch(&mut self, vectors: Vec<Vec<f32>>) -> Vec<u64>

Add multiple vectors

pub fn search(&self, query: &[f32], k: usize) -> Vec<(u64, f32)>

Search for k nearest neighbors

pub fn search_rerank( &self, query: &[f32], k: usize, rerank_k: usize, ) -> Vec<(u64, f32)>

Search with reranking using original vectors

pub fn len(&self) -> usize

Get number of vectors

pub fn is_empty(&self) -> bool

Check if empty

pub fn compression_ratio(&self) -> f32

Get compression ratio

pub fn memory_usage(&self) -> usize

Memory usage in bytes