anofox-ml-linear 0.1.0

Stochastic Gradient Descent linear models (SGDClassifier, SGDRegressor, PassiveAggressive) for anofox-ml
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
//! SGD-based linear models.
//!
//! Provides [`SgdClassifier`] and [`SgdRegressor`] — linear models trained with
//! stochastic gradient descent. These scale well to large datasets and support
//! multiple loss functions and regularization options.

pub mod passive_aggressive;
pub mod sgd_classifier;
pub mod sgd_common;
pub mod sgd_regressor;

pub use passive_aggressive::{
    FittedPassiveAggressiveClassifier, FittedPassiveAggressiveRegressor, PaVariant,
    PassiveAggressiveClassifier, PassiveAggressiveRegressor,
};
pub use sgd_classifier::{ClassifierLoss, FittedSgdClassifier, SgdClassifier};
pub use sgd_common::{LearningRate, Penalty as SgdPenalty};
pub use sgd_regressor::{FittedSgdRegressor, RegressorLoss, SgdRegressor};