Struct MultinomialNB 

pub struct MultinomialNB {
    pub alpha: f32,
    /* private fields */
}

Multinomial Naive Bayes for discrete/count data

Fields

alpha: f32

Implementations

impl MultinomialNB

pub fn new() -> Self

pub fn alpha(self, alpha: f32) -> Self

pub fn fit(&mut self, x: &Tensor, y: &Tensor)

pub fn predict(&self, x: &Tensor) -> Tensor

pub fn score(&self, x: &Tensor, y: &Tensor) -> f32

Trait Implementations

impl Default for MultinomialNB

fn default() -> Self

Returns the “default value” for a type.