average_embedding

Function average_embedding 

Source
pub fn average_embedding(embeddings: &[Vec<f32>]) -> Option<Vec<f32>>
Expand description

Calculate the average embedding from a set of embeddings

Useful for creating centroid embeddings or aggregate representations.

Returns None if the input is empty or embeddings have inconsistent dimensions.

ยงExample

use ipfrs_semantic::utils::average_embedding;

let embeddings = vec![
    vec![1.0, 2.0, 3.0],
    vec![2.0, 3.0, 4.0],
    vec![3.0, 4.0, 5.0],
];

let avg = average_embedding(&embeddings).unwrap();
assert_eq!(avg, vec![2.0, 3.0, 4.0]);