Struct rusty_machine::learning::naive_bayes::Bernoulli

pub struct Bernoulli { /* private fields */ }

The Bernoulli Naive Bayes model distribution.

Defines:

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

Implementations

impl Bernoulli

pub 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, f: &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
) -> 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.