polyvoice 0.17.0

Speaker diarization for Rust — who spoke when. ONNX path optional: default features are empty (ort-free BYO-embedder core); enable onnx for Silero VAD, WeSpeaker embeddings, and Pyannote segmentation.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use super::*;

fn unit(dim: usize, axis: usize) -> Vec<f32> {
    let mut v = vec![0.0_f32; dim];
    v[axis] = 1.0;
    v
}

#[test]
fn compute_centroids_l2_normalized() {
    let embeddings = vec![unit(3, 0), unit(3, 0), unit(3, 1), unit(3, 1)];
    let labels = vec![0, 0, 1, 1];
    let centroids = compute_centroids(&embeddings, &labels);
    assert_eq!(centroids.len(), 2);
    for c in &centroids {
        let n: f32 = c.embedding.iter().map(|x| x * x).sum::<f32>().sqrt();
        assert!(
            (n - 1.0).abs() < 1e-3,
            "centroid not L2-normalized: norm={n}"
        );
    }
}

#[test]
fn compute_centroids_drops_empty_clusters() {
    // Labels skip from 0 to 2; cluster 1 has no members.
    let embeddings = vec![unit(3, 0), unit(3, 1), unit(3, 1)];
    let labels = vec![0, 2, 2];
    let centroids = compute_centroids(&embeddings, &labels);
    assert_eq!(centroids.len(), 2);
    let speakers: Vec<u32> = centroids.iter().map(|c| c.speaker.0).collect();
    assert_eq!(speakers, vec![0, 2]);
}

#[test]
fn compute_centroids_sorted_by_speaker_id() {
    let embeddings = vec![unit(3, 0), unit(3, 1), unit(3, 2)];
    let labels = vec![5, 1, 3];
    let centroids = compute_centroids(&embeddings, &labels);
    let speakers: Vec<u32> = centroids.iter().map(|c| c.speaker.0).collect();
    assert_eq!(speakers, vec![1, 3, 5]);
}

#[test]
fn compute_centroids_empty_input_returns_empty() {
    let centroids = compute_centroids(&[], &[]);
    assert!(centroids.is_empty());
}

#[test]
fn compute_centroids_label_mismatch_returns_empty() {
    // Mismatched lengths: caller bug, conservative empty return rather than panic.
    let centroids = compute_centroids(&[unit(3, 0)], &[0, 1]);
    assert!(centroids.is_empty());
}