Skip to main content

rill_ml/models/
mod.rs

1//! Online models: baseline regressors, linear regression, logistic regression,
2//! Naive Bayes classifiers, and FTRL-Proximal for sparse features.
3
4pub mod baseline;
5pub mod ftrl;
6pub mod linear_regression;
7pub mod logistic_regression;
8pub mod naive_bayes;
9
10pub use baseline::{
11    BaselineConfig, ExponentiallyWeightedMeanRegressor, LastValueRegressor, MeanRegressor,
12};
13pub use ftrl::{FtrlClassifier, FtrlConfig, FtrlRegressor};
14pub use linear_regression::{LinearRegression, LinearRegressionConfig};
15pub use logistic_regression::{LogisticRegression, LogisticRegressionConfig};
16pub use naive_bayes::{
17    BernoulliNaiveBayes, GaussianNaiveBayes, MultinomialNaiveBayes, NaiveBayesConfig,
18};