gathers 0.3.1

Clustering algorithms.
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

use std::path::Path;

use argh::FromArgs;
use env_logger::Env;
use gathers::distance::Distance;
use gathers::kmeans::KMeans;
use gathers::utils::{as_continuous_vec, as_matrix, read_vecs, write_vecs};
use log::debug;

#[derive(FromArgs, Debug)]
/// gathers CLI args
struct Args {
    /// input file path
    #[argh(option, short = 'i')]
    input: String,
    /// output file path
    #[argh(option, short = 'o')]
    output: String,
    /// number of clusters
    #[argh(option, short = 'n', default = "4096")]
    n_cluster: u32,
    /// max number of iterations
    #[argh(option, short = 'm', default = "25")]
    max_iter: u32,
}

fn main() {
    let args: Args = argh::from_env();

    let env = Env::default().filter_or("GATHERS_LOG", "debug");
    env_logger::init_from_env(env);
    debug!("{:?}", args);

    let vecs = read_vecs::<f32>(Path::new(&args.input)).expect("failed to read vecs");
    let dim = vecs[0].len();
    let kmeans = KMeans::new(
        args.n_cluster,
        args.max_iter,
        0.01,
        Distance::SquaredEuclidean,
        false,
    );
    let centroids = kmeans.fit(as_continuous_vec(&vecs), dim);
    let centroids_mat = as_matrix(&centroids, dim);
    write_vecs(Path::new(&args.output), &centroids_mat).expect("failed to write centroids");
}