Struct rusty_machine::learning::naive_bayes::Bernoulli

pub struct Bernoulli {
    // some fields omitted
}

The Bernoulli Naive Bayes model distribution.

Defines:

p(x|C_k) = ∏_i p_k^x_i (1-p)_k^1-x_i

Methods

impl Bernoulli

fn log_probs(&self) -> &Matrix<f64>

The log probability matrix.

A matrix of class by feature model log-probabilities.

Trait Implementations

impl Debug for Bernoulli

fn fmt(&self, __arg_0: &mut Formatter) -> Result

Formats the value using the given formatter.

impl Distribution for Bernoulli

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

Initialize the distribution parameters.

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

Updates the distribution parameters.

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

Compute the joint log likelihood of the data.