feat: BernoulliNB (#31)

* feat: BernoulliNB

* Move preprocessing to a trait in linalg/stats.rs
This commit is contained in:
morenol
2020-12-04 20:45:40 -04:00
committed by GitHub
parent 4720a3a4eb
commit f0b348dd6e
7 changed files with 367 additions and 4 deletions
+2
View File
@@ -64,10 +64,12 @@ impl<T: RealNumber, M: Matrix<T>, D: NBDistribution<T, M>> BaseNaiveBayes<T, M,
Ok(y_hat)
}
}
mod bernoulli;
mod categorical;
mod gaussian;
mod multinomial;
pub use bernoulli::{BernoulliNB, BernoulliNBParameters};
pub use categorical::{CategoricalNB, CategoricalNBParameters};
pub use gaussian::{GaussianNB, GaussianNBParameters};
pub use multinomial::{MultinomialNB, MultinomialNBParameters};