Struct IVFRaBitQIndex 

pub struct IVFRaBitQIndex {
    pub config: IVFConfig,
    pub centroids_version: u64,
    pub random_signs: Vec<i8>,
    pub random_perm: Vec<u32>,
    pub clusters: HashMap<u32, ClusterData>,
    pub num_vectors: usize,
}

IVF-RaBitQ index for a single segment

Fields

config: IVFConfig

Configuration

centroids_version: u64

Version of coarse centroids used (for merge compatibility)

random_signs: Vec<i8>

Random signs for transform (+1 or -1)

random_perm: Vec<u32>

Random permutation for transform

clusters: HashMap<u32, ClusterData>

Cluster data (sparse - only populated clusters)

num_vectors: usize

Total number of vectors indexed

Implementations

impl IVFRaBitQIndex

pub fn new(config: IVFConfig, centroids_version: u64) -> Self

Create a new empty IVF index

pub fn build( config: IVFConfig, coarse_centroids: &CoarseCentroids, vectors: &[Vec<f32>], doc_ids: Option<&[u32]>, ) -> Self

Build index from vectors using provided coarse centroids

pub fn add_vector( &mut self, coarse_centroids: &CoarseCentroids, doc_id: u32, vector: &[f32], )

Add a single vector to the index

pub fn search( &self, coarse_centroids: &CoarseCentroids, query: &[f32], k: usize, nprobe: usize, ) -> Vec<(u32, f32)>

Search for k nearest neighbors

pub fn merge( indexes: &[&IVFRaBitQIndex], doc_id_offsets: &[u32], ) -> Result<Self, &'static str>

Merge multiple IVF indexes (O(1) per cluster - just concatenate)

pub fn num_clusters(&self) -> usize

Get number of populated clusters

pub fn len(&self) -> usize

Get total number of vectors

pub fn is_empty(&self) -> bool

pub fn size_bytes(&self) -> usize

Approximate size in bytes

Trait Implementations

impl Clone for IVFRaBitQIndex

fn clone(&self) -> IVFRaBitQIndex

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for IVFRaBitQIndex

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for IVFRaBitQIndex

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for IVFRaBitQIndex

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.