Function improve_clusters

crates/openquant/src/onc.rs:191–219

(
    corr_mat: &DMatrix<f64>,
    kept_clusters: &BTreeMap<usize, Vec<usize>>,
    top_clusters: &BTreeMap<usize, Vec<usize>>,
)

}

fn improve_clusters(
    corr_mat: &DMatrix<f64>,
    kept_clusters: &BTreeMap<usize, Vec<usize>>,
    top_clusters: &BTreeMap<usize, Vec<usize>>,
) -> Result<ClusterState, OncError> {
    let mut clusters_new: BTreeMap<usize, Vec<usize>> = BTreeMap::new();
    for members in kept_clusters.values() {
        clusters_new.insert(clusters_new.len(), members.clone());
    }
    for members in top_clusters.values() {
        clusters_new.insert(clusters_new.len(), members.clone());
    }

    let mut new_idx = Vec::new();
    for members in clusters_new.values() {
        new_idx.extend(members.iter().copied());
    }

    let corr_new = submatrix(corr_mat, &new_idx);
    let labels = labels_from_clusters(corr_mat.nrows(), &clusters_new);
    let dist = corr_to_distance(corr_mat);
    let silh_scores_new = silhouette_samples(&dist, &labels);

    Ok(ClusterState {
        ordered_correlation: corr_new,
        clusters: clusters_new,
        silhouette_scores: silh_scores_new,
    })
}

fn cluster_kmeans_base(
    corr_mat: &DMatrix<f64>,