feat: BernoulliNB (#31)
* feat: BernoulliNB * Move preprocessing to a trait in linalg/stats.rs
This commit is contained in:
@@ -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};
|
||||
|
||||
Reference in New Issue
Block a user