torsh-cluster 0.1.2

Unsupervised learning and clustering algorithms for ToRSh, powered by SciRS2
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

//! Initialization strategies for clustering algorithms

pub mod forgy;
pub mod kmeans_plus_plus;
pub mod random_partition;

pub use forgy::Forgy;
pub use kmeans_plus_plus::KMeansPlusPlus;
pub use random_partition::RandomPartition;

use crate::error::ClusterResult;
use torsh_tensor::Tensor;

/// Trait for initialization strategies
pub trait InitializationStrategy {
    /// Initialize cluster centers
    fn initialize(
        &self,
        data: &Tensor,
        n_clusters: usize,
        seed: Option<u64>,
    ) -> ClusterResult<Tensor>;

    /// Get strategy name
    fn name(&self) -> &str;
}