linfa-hierarchical 0.8.1

Agglomerative Hierarchical clustering
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

use std::error::Error;

use linfa::traits::Transformer;
use linfa_hierarchical::HierarchicalCluster;
use linfa_kernel::{Kernel, KernelMethod};

fn main() -> Result<(), Box<dyn Error>> {
    // load Iris plant dataset
    let dataset = linfa_datasets::iris();

    let kernel = Kernel::params()
        .method(KernelMethod::Gaussian(1.0))
        .transform(dataset.records().view());

    let kernel = HierarchicalCluster::default()
        .num_clusters(3)
        .transform(kernel)?;

    for (id, target) in kernel.targets().iter().zip(dataset.targets().into_iter()) {
        let name = match *target {
            0 => "setosa",
            1 => "versicolor",
            2 => "virginica",
            _ => unreachable!(),
        };

        print!("({id} {name}) ");
    }
    println!();

    Ok(())
}