Struct lance_index::vector::ivf::IvfImpl

pub struct IvfImpl<T: ArrowFloatType + Dot + L2 + Cosine> { /* private fields */ }

IVF - IVF file partition

Implementations

impl<T: ArrowFloatType + Dot + L2 + Cosine + 'static> IvfImpl<T>

pub fn new( centroids: MatrixView<T>, metric_type: MetricType, transforms: Vec<Arc<dyn Transformer>>, range: Option<Range<u32>>, precomputed_partitions: Option<HashMap<u64, u32>> ) -> Self

Trait Implementations

impl<T: Clone + ArrowFloatType + Dot + L2 + Cosine> Clone for IvfImpl<T>

fn clone(&self) -> IvfImpl<T>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<T: Debug + ArrowFloatType + Dot + L2 + Cosine> Debug for IvfImpl<T>

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

Formats the value using the given formatter.

impl<T: ArrowFloatType + Dot + L2 + Cosine + 'static> Ivf for IvfImpl<T>

fn compute_partitions<'life0, 'life1, 'async_trait>( &'life0 self, data: &'life1 FixedSizeListArray ) -> Pin<Box<dyn Future<Output = Result<UInt32Array>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Compute the partitions for each vector in the input data.

fn compute_residual<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, original: &'life1 FixedSizeListArray, partitions: Option<&'life2 UInt32Array> ) -> Pin<Box<dyn Future<Output = Result<FixedSizeListArray>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Compute residual vector.

fn find_partitions( &self, query: &dyn Array, nprobes: usize ) -> Result<UInt32Array>

Find the closest partitions for the query vector.

fn partition_transform<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, batch: &'life1 RecordBatch, column: &'life2 str ) -> Pin<Box<dyn Future<Output = Result<RecordBatch>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Partition a batch of vectors into multiple batches, each batch contains vectors and other data.