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(crate) mod baseline;
5pub(crate) mod ftrl;
6pub(crate) mod linear_regression;
7pub(crate) mod logistic_regression;
8pub(crate) mod naive_bayes;
9
10pub use baseline::{
11    BaselineConfig, ExponentiallyWeightedMeanRegressor, LastValueRegressor, MeanRegressor,
12};
13pub use ftrl::{FtrlClassifier, FtrlConfig, FtrlRegressor, NewFeaturePolicy};
14pub use linear_regression::{LinearRegression, LinearRegressionConfig};
15pub use logistic_regression::{LogisticRegression, LogisticRegressionConfig};
16pub use naive_bayes::{
17    BernoulliNaiveBayes, GaussianNaiveBayes, MultinomialNaiveBayes, NaiveBayesConfig,
18};