Skip to content

Instantly share code, notes, and snippets.

View guimarqu's full-sized avatar

Guillaume Marques guimarqu

  • Bordeaux, France
  • 17:30 (UTC +02:00)
  • LinkedIn in/guimarqu
View GitHub Profile
@guimarqu
guimarqu / agglomerative_clustering_max_dist.rs
Created March 14, 2024 12:56
Agglomerative clustering with precomputed metric, complete linkage, and maximum distance threshold
// Adapted from https://docs.rs/linfa-hierarchical/latest/src/linfa_hierarchical/lib.rs.html
pub fn agglomerative_clustering(
dist_matrix: &mut [f64],
num_observations: usize,
max_dist: f64,
) -> anyhow::Result<Vec<i32>> {
let res = linkage(dist_matrix, num_observations, kodama::Method::Complete);
let mut clusters = (0..num_observations)