Struct KMeansModel 

pub struct KMeansModel {
    pub centroids: Array2<f64>,
    pub n_clusters: usize,
    pub n_iter: usize,
    pub inertia: f64,
    pub labels: Option<Array1<usize>>,
}

K-means clustering model that can be serialized

Fields

centroids: Array2<f64>

Cluster centroids

n_clusters: usize

Number of clusters

n_iter: usize

Number of iterations performed

inertia: f64

Sum of squared distances

labels: Option<Array1<usize>>

Cluster labels for training data (optional)

Implementations

impl KMeansModel

pub fn new( centroids: Array2<f64>, n_clusters: usize, n_iter: usize, inertia: f64, labels: Option<Array1<usize>>, ) -> Self

Create a new K-means model

pub fn predict(&self, data: ArrayView2<'_, f64>) -> Result<Array1<usize>>

Predict cluster labels for new data

pub fn predict_single(&self, point: &[f64]) -> Result<usize>

Get the closest cluster center for a single point

Trait Implementations

impl AdvancedExport for KMeansModel

Implementation of AdvancedExport for KMeansModel

fn export_with_metadata( &self, format: ExportFormat, metadata: Option<ModelMetadata>, ) -> Result<Vec<u8>>

Export model to specified format with metadata

fn export_to_file<P: AsRef<Path>>(&self, path: P) -> Result<()>

Export to file with automatic format detection

fn export_summary(&self) -> Result<String>

Export model summary for quick inspection

fn export_compatible(&self, target_library: &str) -> Result<Value>

Export model in a format compatible with other libraries

fn validate_for_export(&self) -> Result<()>

Validate model before export

impl Clone for KMeansModel

fn clone(&self) -> KMeansModel

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl ClusteringModel for KMeansModel

fn n_clusters(&self) -> usize

Get the number of clusters

fn predict(&self, data: ArrayView2<'_, f64>) -> Result<Array1<usize>>

Predict cluster labels for new data

fn summary(&self) -> Result<Value>

Get model summary as JSON

impl Debug for KMeansModel

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for KMeansModel

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl SerializableModel for KMeansModel

fn save_to_file<P: AsRef<Path>>(&self, path: P) -> Result<()>

Save the model to a file

fn save_to_writer<W: Write>(&self, writer: W) -> Result<()>

Save the model to a writer

fn save_to_file_compressed<P: AsRef<Path>>(&self, path: P) -> Result<()>

Save the model to a file with compression

fn load_from_file_compressed<P: AsRef<Path>>(path: P) -> Result<Self>

Load the model from a compressed file

fn load_from_file<P: AsRef<Path>>(path: P) -> Result<Self>

Load the model from a file

fn load_from_reader<R: Read>(reader: R) -> Result<Self>

Load the model from a reader

impl Serialize for KMeansModel

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.