Struct KMeans 

pub struct KMeans {
    pub n_clusters: usize,
    pub max_iter: usize,
    pub tol: f32,
    pub n_init: usize,
    pub init: KMeansInit,
    pub cluster_centers_: Option<Vec<Vec<f32>>>,
    pub labels_: Option<Vec<usize>>,
    pub inertia_: Option<f32>,
    pub n_iter_: usize,
}

K-Means clustering using Lloyd’s algorithm

Fields

n_clusters: usize

max_iter: usize

tol: f32

n_init: usize

init: KMeansInit

cluster_centers_: Option<Vec<Vec<f32>>>

labels_: Option<Vec<usize>>

inertia_: Option<f32>

n_iter_: usize

Implementations

impl KMeans

pub fn new(n_clusters: usize) -> Self

pub fn max_iter(self, n: usize) -> Self

pub fn n_init(self, n: usize) -> Self

pub fn init(self, init: KMeansInit) -> Self

pub fn fit(&mut self, x: &Tensor)

pub fn predict(&self, x: &Tensor) -> Tensor

pub fn fit_predict(&mut self, x: &Tensor) -> Tensor

pub fn transform(&self, x: &Tensor) -> Tensor