gathers

Clustering algorithm implementation in Rust and binding to Python.

For Python users, check the Python README.

• K-means
• PyO3 binding
• RaBitQ assignment
• Parallel with Rayon
• x86 & x86_64 SIMD acceleration
• mini batch K-means
• Hierarchical K-means
• arm & aarch64 SIMD acceleration

Installation

cargo add gathers

Usage

Check the docs and main.rs.

lib.rs:

Clustering algorithms for Rust.

Examples

use gathers::kmeans::{KMeans, rabitq_assign};
use gathers::utils::as_continuous_vec;


let kmeans = KMeans::default();
let num = vecs.len();
let dim = vecs[0].len();

// fit
let centroids = kmeans.fit(as_continuous_vec(&vecs), dim);
// predict
let mut labels = vec![0; num];
rabitq_assign(&as_continuous_vec(&vecs), &centroids, dim, &mut labels);