Struct PqCodebook 

pub struct PqCodebook {
    pub centroids: Vec<Vec<Vec<f32>>>,
    pub num_subvectors: usize,
    pub subvector_dim: usize,
}

A simple product quantization codebook.

Each subvector has 256 centroids. During encoding, each subvector is mapped to its nearest centroid. During decoding, the centroid values are concatenated to reconstruct the approximation.

Fields

centroids: Vec<Vec<Vec<f32>>>

Centroids indexed by [subvector_index][centroid_index]. Each centroid is a flat vector of subvector_dim floats.

num_subvectors: usize

Number of subvectors.

subvector_dim: usize

Dimension per subvector.

Implementations

impl PqCodebook

pub fn build(vectors: &[Vec<f32>], dim: usize) -> Self

Build a trivial codebook from a set of vectors.

This is a simplified implementation that uses uniform quantization (divides the range into 256 equal bins per dimension). A production implementation would use k-means clustering.

pub fn encode(&self, vector: &[f32]) -> PqVector

Encode a vector into PQ codes.

pub fn decode(&self, pq: &PqVector) -> Vec<f32>

Decode PQ codes back to an approximate vector.

Trait Implementations

impl Clone for PqCodebook

fn clone(&self) -> PqCodebook

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PqCodebook

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for PqCodebook

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for PqCodebook

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

where __S: Serializer,

Serialize this value into the given Serde serializer.