Struct MultinomialNaiveBayes 

pub struct MultinomialNaiveBayes { /* private fields */ }

Multinomial Naive Bayes classifier for text

Suitable for text classification with word count / TF-IDF features. Implements Laplace smoothing.

Implementations

impl MultinomialNaiveBayes

pub fn new(alpha: f64) -> Self

Create a new multinomial Naive Bayes classifier

pub fn fit(&mut self, features: &Array2<f64>, labels: &[String]) -> Result<()>

Train the classifier

Arguments

• features - Feature matrix (n_samples x n_features), e.g. TF-IDF
• labels - Class labels for each sample

pub fn predict(&self, features: &Array2<f64>) -> Result<Vec<String>>

Predict class labels for feature matrix

Trait Implementations

impl Clone for MultinomialNaiveBayes

fn clone(&self) -> MultinomialNaiveBayes

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for MultinomialNaiveBayes

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

Formats the value using the given formatter.