Function assign

nodedb-vector/src/multivec/plaid.rs:25–38

Assign each vector to its nearest centroid index.

(vectors: &[Vec<f32>], centroids: &[Vec<f32>])

/// Assign each vector to its nearest centroid index.
fn assign(vectors: &[Vec<f32>], centroids: &[Vec<f32>]) -> Vec<usize> {
    vectors
        .iter()
        .map(|v| {
            centroids
                .iter()
                .enumerate()
                .map(|(i, c)| (i, scalar_distance(v, c, DistanceMetric::L2)))
                .min_by(|a, b| a.1.partial_cmp(&b.1).unwrap_or(std::cmp::Ordering::Equal))
                .map(|(i, _)| i)
                .unwrap_or(0)
        })
        .collect()
}

/// Deterministic LCG step for reproducible randomness across train calls.
fn lcg_next(s: &mut u64) -> u64 {