[][src]Struct naivebayes::NaiveBayes

pub struct NaiveBayes { /* fields omitted */ }

Methods

impl NaiveBayes[src]

pub fn new() -> NaiveBayes[src]

creates a new instance of a NaiveBayes classifier.

pub fn train(&mut self, data: &Vec<String>, label: &String)[src]

trains the model with a Vec<String> of tokens, associating it with a String label.

pub fn classify(&mut self, data: &Vec<String>) -> HashMap<String, f64>[src]

classify a Vec<String> of tokens returning a map of tokens and probabilities as keys and values, respectively.

pub fn log_classify(&mut self, data: &Vec<String>) -> HashMap<String, f64>[src]

classify a Vec<String> of tokens returning a map of tokens and log-probabilities as keys and values, respectively. Using log_classify may prevent underflows.

Auto Trait Implementations

impl Send for NaiveBayes

impl Sync for NaiveBayes

Blanket Implementations

impl<T> From for T[src]

impl<T, U> Into for T where
    U: From<T>, 
[src]

impl<T, U> TryFrom for T where
    U: Into<T>, 
[src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T> Borrow for T where
    T: ?Sized
[src]

impl<T> Any for T where
    T: 'static + ?Sized
[src]

impl<T> BorrowMut for T where
    T: ?Sized
[src]

impl<T, U> TryInto for T where
    U: TryFrom<T>, 
[src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.