Function is_diverse_batched

nodedb-vector/src/batch_distance.rs:31–44

Precompute all pairwise distances between selected neighbors and a candidate. For the diversity heuristic: given a candidate and the currently selected set, compute `distance(candidate, selected[i])` for all i. Returns true if the candidate is "diverse" (closer to query than to every selected neighbor).

(
    candidate_vec: &[f32],
    candidate_dist_to_query: f32,
    selected_vecs: &[&[f32]],
    metric: DistanceMetric,
)

/// Returns true if the candidate is "diverse" (closer to query than to
/// every selected neighbor).
pub fn is_diverse_batched(
    candidate_vec: &[f32],
    candidate_dist_to_query: f32,
    selected_vecs: &[&[f32]],
    metric: DistanceMetric,
) -> bool {
    for selected in selected_vecs {
        let dist_to_selected = distance(candidate_vec, selected, metric);
        if candidate_dist_to_query > dist_to_selected {
            return false;
        }
    }
    true
}

#[cfg(test)]
mod tests {