lance::index::vector::pq

Function build_pq_model

Source 
pub async fn build_pq_model(
    dataset: &Dataset,
    column: &str,
    dim: usize,
    metric_type: MetricType,
    params: &PQBuildParams,
    ivf: Option<&IvfModel>,
) -> Result<ProductQuantizer>
Expand description

Train Product Quantizer model.

Parameters:

  • dataset: The dataset to train the PQ model.
  • column: The column name of the dataset.
  • dim: The dimension of the vectors.
  • metric_type: The metric type of the vectors.
  • params: The parameters to train the PQ model.
  • ivf: If provided, the IVF model to compute the residual for PQ training.