naivebayes

Crates.ionaivebayes
lib.rsnaivebayes
version0.1.2
sourcesrc
created_at2018-11-28 19:42:22.059579
updated_at2019-03-25 18:39:48.159074
descriptionA simple Naive Bayes classifier.
homepagehttps://gitlab.com/ruivieira/naive-bayes
repositoryhttps://gitlab.com/ruivieira/naive-bayes
max_upload_size
id99122
size27,663
Rui Vieira (ruivieira)

documentation

README

https://crates.io/crates/naivebayes https://docs.rs/naivebayes/

naive-bayes

A Naive Bayes classifier written in Rust.

installation

Add to your Cargo.toml:

naivebayes = "0.1.1"

usage

Add the crate and NaiveBayes to your code:

extern crate naivebayes;

use naivebayes::NaiveBayes;

Initialise the classifier and train it classifier by passing Vec<String> of tokens, along with a label:

let mut nb = NaiveBayes::new();
nb.train(&tokens, &label);

Use another set of tokens as Vec<String> to classify it:

let classification = nb.classify(&tokens_classify);
print!("classification = {:?}", classification);

Alternitavely, to prevent a potential calculation underflow with very small probabilities, the log_classify method can be used:

let classification = nb.log_classify(&tokens_classify);
print!("classification = {:?}", classification);
Commit count: 23

cargo fmt