polyvoice 0.6.9

Speaker diarization for Rust — who spoke when. ONNX-powered: Silero VAD, WeSpeaker embeddings, Pyannote segmentation, K-means/AHC clustering, overlap detection.
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
56
57
58
59
60

#![allow(clippy::unwrap_used)]
//! Integration test for `AhcClusterer` and `NmeScClusterer` on synthetic
//! clusters. Pure-CPU; runs in normal `cargo test` (no network or model required).

#![cfg(feature = "clusterer")]

use polyvoice::clusterer::{AhcClusterer, Clusterer};

#[cfg(feature = "spectral")]
use polyvoice::clusterer::NmeScClusterer;

fn synth_clusters_4(d: usize) -> Vec<Vec<f32>> {
    let centers: Vec<Vec<f32>> = (0..4)
        .map(|c| {
            let mut v = vec![0.0_f32; d];
            v[c] = 1.0;
            v
        })
        .collect();
    let mut all = Vec::new();
    for _ in 0..6 {
        for c in &centers {
            let mut perturbed = c.clone();
            perturbed[0] += 0.01;
            let n: f32 = perturbed.iter().map(|x| x * x).sum::<f32>().sqrt();
            for x in &mut perturbed {
                *x /= n;
            }
            all.push(perturbed);
        }
    }
    all
}

#[test]
fn ahc_finds_four_clusters() {
    let c = AhcClusterer::default();
    let labels = c.cluster(&synth_clusters_4(8)).unwrap();
    let unique: std::collections::HashSet<usize> = labels.iter().copied().collect();
    assert!(
        (3..=5).contains(&unique.len()),
        "got {} clusters: {:?}",
        unique.len(),
        labels
    );
}

#[cfg(feature = "spectral")]
#[test]
fn nme_sc_finds_four_clusters() {
    let c = NmeScClusterer::default();
    let labels = c.cluster(&synth_clusters_4(8)).unwrap();
    let unique: std::collections::HashSet<usize> = labels.iter().copied().collect();
    assert!(
        (3..=5).contains(&unique.len()),
        "got {} clusters: {:?}",
        unique.len(),
        labels
    );
}