Struct GMMModel 

pub struct GMMModel {
    pub weights: Array1<f64>,
    pub means: Array2<f64>,
    pub covariances: Vec<Array2<f64>>,
    pub n_components: usize,
    pub covariance_type: String,
    pub log_likelihood: f64,
    pub converged: bool,
    pub n_iter: usize,
}

Gaussian Mixture Model that can be serialized

Fields

weights: Array1<f64>

Mixture weights

means: Array2<f64>

Component means

covariances: Vec<Array2<f64>>

Component covariances

n_components: usize

Number of components

covariance_type: String

Covariance type

log_likelihood: f64

Log-likelihood

converged: bool

Converged flag

n_iter: usize

Number of iterations

Implementations

impl GMMModel

pub fn new( weights: Array1<f64>, means: Array2<f64>, covariances: Vec<Array2<f64>>, n_components: usize, covariance_type: String, log_likelihood: f64, converged: bool, n_iter: usize, ) -> Self

Create a new GMM model

pub fn predict_proba(&self, data: ArrayView2<'_, f64>) -> Result<Array2<f64>>

Predict cluster probabilities for new data

Trait Implementations

impl Clone for GMMModel

fn clone(&self) -> GMMModel

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GMMModel

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for GMMModel

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl SerializableModel for GMMModel

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 GMMModel

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

where __S: Serializer,

Serialize this value into the given Serde serializer.