Struct RaBitQCodebook 

pub struct RaBitQCodebook {
    pub config: RaBitQConfig,
    pub random_signs: Vec<i8>,
    pub random_perm: Vec<u32>,
    pub version: u64,
}

RaBitQ codebook (random transform parameters)

Trained once, shared across all segments for merge compatibility.

Fields

config: RaBitQConfig

Configuration

random_signs: Vec<i8>

Random signs for transform (+1 or -1)

random_perm: Vec<u32>

Random permutation for transform

version: u64

Version for merge compatibility checking

Implementations

impl RaBitQCodebook

pub fn new(config: RaBitQConfig) -> Self

Create a new RaBitQ codebook with random transform

pub fn encode( &self, vector: &[f32], centroid: Option<&[f32]>, ) -> QuantizedVector

Encode a vector to binary quantized form

If centroid is provided, encodes the residual (vector - centroid).

pub fn prepare_query( &self, query: &[f32], centroid: Option<&[f32]>, ) -> QuantizedQuery

Prepare a query for fast distance estimation

pub fn estimate_distance( &self, query: &QuantizedQuery, code: &QuantizedVector, ) -> f32

Estimate squared distance between query and a quantized vector

pub fn size_bytes(&self) -> usize

Memory usage in bytes

Trait Implementations

impl Clone for RaBitQCodebook

fn clone(&self) -> RaBitQCodebook

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for RaBitQCodebook

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for RaBitQCodebook

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Quantizer for RaBitQCodebook

type Code = QuantizedVector

The quantized code type

type Config = RaBitQConfig

Configuration type

type QueryData = QuantizedQuery

Query-specific precomputed data for fast distance computation

fn encode(&self, vector: &[f32], centroid: Option<&[f32]>) -> Self::Code

Encode a vector (optionally relative to a centroid)

fn prepare_query( &self, query: &[f32], centroid: Option<&[f32]>, ) -> Self::QueryData

Prepare query-specific data for fast distance computation

fn compute_distance( &self, query_data: &Self::QueryData, code: &Self::Code, ) -> f32

Compute approximate distance using precomputed query data

fn size_bytes(&self) -> usize

Memory usage of the quantizer itself (codebooks, etc.)

fn decode(&self, _code: &Self::Code) -> Option<Vec<f32>>

Decode a code back to approximate vector (if supported)

impl Serialize for RaBitQCodebook

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

where __S: Serializer,

Serialize this value into the given Serde serializer.