Struct rust_nb::Model

pub struct Model<T: ModelStore + Sync> { /* fields omitted */ }

Methods

impl<T: ModelStore + Sync> Model<T>

pub fn with_stop_words_file(self, stop_words_file: &str) -> Self

pub fn with_default_gaussian_m2(self, default_gaussian_m2: f64) -> Self

pub fn with_pseudo_count(self, pseudo_count: f64) -> Self

pub fn with_prior_factor(self, prior_factor: f64) -> Self

pub fn with_default_gaussian_sigma_factor(
    self,
    default_gaussian_sigma_factor: f64
) -> Self

pub fn train(
    &mut self,
    model_name: &str,
    class_feature_pairs: &[(String, Vec<Feature>)]
)

pub fn predict(
    &self,
    model_name: &str,
    features: &[Feature]
) -> HashMap<String, f64>

pub fn predict_batch(
    &self,
    model_name: &str,
    features_vec: &[Vec<Feature>]
) -> Vec<HashMap<String, f64>>

impl Model<ModelHashMapStore>

pub fn new() -> Model<ModelHashMapStore>

Trait Implementations

impl<'de, T: ModelStore + Sync> Deserialize<'de> for Model<T> where
    T: Deserialize<'de>, 

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
    __D: Deserializer<'de>, 

Deserialize this value from the given Serde deserializer.

impl<T: ModelStore + Sync> Serialize for Model<T> where
    T: Serialize, 

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error> where
    __S: Serializer, 

Serialize this value into the given Serde serializer.