Struct Codebook 

pub struct Codebook {
    pub config: CodebookConfig,
    pub centroids: Vec<f32>,
    pub id: String,
    pub stats: CodebookStats,
}

A codebook containing centroids for vector quantization

Fields

config: CodebookConfig

Configuration

centroids: Vec<f32>

Centroid vectors [num_centroids × centroid_dim]

id: String

Codebook identifier

stats: CodebookStats

Training statistics

Implementations

impl Codebook

pub fn new(config: CodebookConfig, id: impl Into<String>) -> Self

Create a new codebook with random centroids

pub fn from_centroids( config: CodebookConfig, centroids: Vec<f32>, id: impl Into<String>, ) -> Result<Self>

Create from existing centroids

pub fn find_nearest(&self, vector: &[f32]) -> Result<(usize, f32)>

Find nearest centroid for a vector

pub fn find_k_nearest( &self, vector: &[f32], k: usize, ) -> Result<Vec<(usize, f32)>>

Find K nearest centroids

pub fn get_centroid(&self, index: usize) -> Option<&[f32]>

Get centroid by index

pub fn encode_batch( &self, vectors: &[f32], vector_dim: usize, ) -> Result<Vec<u16>>

Encode a batch of vectors to indices

pub fn decode_batch(&self, indices: &[u16]) -> Vec<f32>

Decode indices to vectors

pub fn to_bytes(&self) -> Vec<u8>

Serialize to bytes

pub fn from_bytes(bytes: &[u8]) -> Result<Self>

Deserialize from bytes

Trait Implementations

impl Clone for Codebook

fn clone(&self) -> Codebook

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Codebook

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Codebook

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for Codebook

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

where __S: Serializer,

Serialize this value into the given Serde serializer.