Struct rusty_machine::learning::naive_bayes::Gaussian

pub struct Gaussian { /* private fields */ }

The Gaussian Naive Bayes model distribution.

Defines:

p(x|C_k) = ∏_i N(x_i ; μ_k, σ²_k)

Implementations

impl Gaussian

pub fn theta(&self) -> &Matrix<f64>

Returns the distribution means.

This is a matrix of class by feature means.

pub fn sigma(&self) -> &Matrix<f64>

Returns the distribution variances.

This is a matrix of class by feature variances.

Trait Implementations

impl Debug for Gaussian

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

Formats the value using the given formatter.

impl Distribution for Gaussian

fn from_model_params(class_count: usize, features: usize) -> Gaussian

Initialize the distribution parameters.

fn update_params(
    &mut self,
    data: &Matrix<f64>,
    class: usize
) -> LearningResult<()>

Updates the distribution parameters.

fn joint_log_lik(
    &self,
    data: &Matrix<f64>,
    class_prior: &[f64]
) -> LearningResult<Matrix<f64>>

Compute the joint log likelihood of the data.