anofox-ml-regression 0.1.0

Classical regression models (OLS, Ridge, Lasso, Elastic Net, GLM, GP) for anofox-ml
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

//! Classical regression models wrapping the `anofox-regression` crate.
//!
//! Provides OLS, Ridge, Lasso, Elastic Net, WLS, Quantile, Isotonic, and GLM
//! (Poisson, Binomial) regressors that implement the anofox-ml `Fit` / `Predict`
//! type-state pattern.
//!
//! Cross-validated variants (`RidgeCrossValidated`, `LassoCrossValidated`,
//! `ElasticNetCrossValidated`) automatically select the best regularization
//! parameters using k-fold cross-validation.
//!
//! All models operate on `f64` (not generic over `Float`) because the
//! underlying `anofox-regression` crate only supports `f64`.

pub mod bayesian_ridge;
pub mod convert;
pub mod elastic_net;
pub mod elastic_net_cv;
pub mod glm;
pub mod huber;
pub mod isotonic;
pub mod kernel_ridge;
pub mod lars;
pub mod lasso;
pub mod lasso_cv;
pub mod logistic;
pub mod ols;
pub mod omp;
pub mod quantile;
pub mod ridge;
pub mod ridge_cv;
pub mod robust;
pub mod transformed_target;
pub mod tweedie;
pub mod wls;

pub use bayesian_ridge::{ARDRegression, BayesianRidge, FittedARDRegression, FittedBayesianRidge};
pub use elastic_net::{ElasticNetRegressor, FittedElasticNetRegressor};
pub use elastic_net_cv::{ElasticNetCrossValidated, FittedElasticNetCrossValidated};
pub use glm::{
    BinomialRegressor, FittedBinomialRegressor, FittedPoissonRegressor, PoissonRegressor,
};
pub use huber::{FittedHuberRegressor, HuberRegressor};
pub use isotonic::{FittedIsotonicRegressor, IsotonicRegressor};
pub use kernel_ridge::{FittedKernelRidge, KernelRidge};
pub use lars::{FittedLars, FittedLassoLarsIC, IcCriterion, Lars, LassoLarsIC};
pub use lasso::{FittedLassoRegressor, LassoRegressor};
pub use lasso_cv::{FittedLassoCrossValidated, LassoCrossValidated};
pub use logistic::{FittedLogisticRegressor, LogisticRegressor};
pub use ols::{FittedOlsRegressor, OlsRegressor};
pub use omp::{FittedOrthogonalMatchingPursuit, OrthogonalMatchingPursuit};
pub use quantile::{FittedQuantileRegressor, QuantileRegressor};
pub use ridge::{FittedRidgeRegressor, FittedWeightedRidgeRegressor, RidgeRegressor};
pub use ridge_cv::{FittedRidgeCrossValidated, RidgeCrossValidated};
pub use robust::{
    FittedRansacRegressor, FittedTheilSenRegressor, RansacRegressor, TheilSenRegressor,
};
pub use transformed_target::{FittedTransformedTargetRegressor, TransformedTargetRegressor};
pub use tweedie::{
    gamma_regressor, FittedGammaRegressor, FittedTweedieRegressor, GammaRegressor, TweedieRegressor,
};
pub use wls::{FittedWlsRegressor, WlsRegressor};