Neighbors Ivf Pq Params Module

View as Markdown

Rust module: cuvs::neighbors::ivf_pq::params

Source: rust/cuvs/src/neighbors/ivf_pq/params.rs

Builder-pattern parameter types for IVF-PQ build and search.

All setters are optional; unset values retain the library defaults from the underlying C *ParamsCreate functions.

IndexParams

1pub struct IndexParams {
2 /* private fields */
3}

Parameters for building an IVF-PQ index.

Methods

NameSource
newrust/cuvs/src/neighbors/ivf_pq/params.rs:29

new

1#[builder]
2#[allow(clippy::too_many_arguments)]
3pub fn new(
4n_lists: Option<u32>,
5metric: Option<DistanceType>,
6kmeans_n_iters: Option<u32>,
7kmeans_trainset_fraction: Option<f64>,
8pq_bits: Option<u32>,
9pq_dim: Option<u32>,
10codebook_kind: Option<CodebookGen>,
11codes_layout: Option<ListLayout>,
12force_random_rotation: Option<bool>,
13max_train_points_per_pq_code: Option<u32>,
14add_data_on_build: Option<bool>,
15) -> Result<Self, IvfPqError>

Source: rust/cuvs/src/neighbors/ivf_pq/params.rs:29

Source: rust/cuvs/src/neighbors/ivf_pq/params.rs:21

SearchParams

1pub struct SearchParams {
2 /* private fields */
3}

Parameters for searching an IVF-PQ index.

Methods

NameSource
newrust/cuvs/src/neighbors/ivf_pq/params.rs:143

new

1#[builder]
2pub fn new(
3n_probes: Option<u32>,
4lut_dtype: Option<LutDType>,
5internal_distance_dtype: Option<InternalDistanceDType>,
6) -> Result<Self, IvfPqError>

Source: rust/cuvs/src/neighbors/ivf_pq/params.rs:143

Source: rust/cuvs/src/neighbors/ivf_pq/params.rs:136