Struct rusty_machine::learning::naive_bayes::Bernoulli

pub struct Bernoulli { /* 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)
                 -> 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.