Struct ProductQuantizer 

pub struct ProductQuantizer {
    pub m: usize,
    pub dsub: usize,
    pub dim: usize,
    pub centroids: Vec<Vec<Vec<f32>>>,
    pub trained: bool,
}

Product Quantizer with M subspaces, 256 centroids each (1 byte per subspace)

Fields

m: usize

Number of subspaces

dsub: usize

Dimensions per subspace

dim: usize

Total dimensions

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

Centroids: [m][256][dsub]

trained: bool

Whether the PQ has been trained

Implementations

impl ProductQuantizer

pub fn new(dim: usize, m: usize) -> Result<Self>

Create a new PQ with M subspaces for D-dimensional vectors

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

Train PQ centroids using k-means on training vectors

pub fn encode(&self, vector: &[f32]) -> Result<Vec<u8>>

Encode a vector into PQ codes (M bytes)

pub fn build_distance_table(&self, query: &[f32]) -> Result<Vec<f32>>

Build flat asymmetric distance table for a query vector Returns flat table[subspace * 256 + centroid_id] = distance

pub fn distance_with_table(&self, codes: &[u8], table: &[f32]) -> f32

Compute approximate distance using flat precomputed table

Trait Implementations

impl<'__de, __Context> BorrowDecode<'__de, __Context> for ProductQuantizer

fn borrow_decode<__D: BorrowDecoder<'__de, Context = __Context>>( decoder: &mut __D, ) -> Result<Self, DecodeError>

Attempt to decode this type with the given BorrowDecode.

impl Clone for ProductQuantizer

fn clone(&self) -> ProductQuantizer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<__Context> Decode<__Context> for ProductQuantizer

fn decode<__D: Decoder<Context = __Context>>( decoder: &mut __D, ) -> Result<Self, DecodeError>

Attempt to decode this type with the given Decode.

impl<'de> Deserialize<'de> for ProductQuantizer

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Encode for ProductQuantizer

fn encode<__E: Encoder>(&self, encoder: &mut __E) -> Result<(), EncodeError>

Encode a given type.

impl Serialize for ProductQuantizer

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

where __S: Serializer,

Serialize this value into the given Serde serializer.