MultinomialNB (#32)

feat: add MultinomialNB
This commit is contained in:
morenol
2020-12-03 09:51:33 -04:00
committed by GitHub
parent c172c407d2
commit 4720a3a4eb
2 changed files with 281 additions and 0 deletions
+3
View File
@@ -66,5 +66,8 @@ impl<T: RealNumber, M: Matrix<T>, D: NBDistribution<T, M>> BaseNaiveBayes<T, M,
}
mod categorical;
mod gaussian;
mod multinomial;
pub use categorical::{CategoricalNB, CategoricalNBParameters};
pub use gaussian::{GaussianNB, GaussianNBParameters};
pub use multinomial::{MultinomialNB, MultinomialNBParameters};