Skip to main content

Module optim

Module optim 

Source
Expand description

Optimizers for online linear models.

Optimizers are represented as a concrete enum to avoid trait-object overhead and simplify serialization. The internal parameter vector has length feature_count + 1, where the last position holds the intercept.

Re-exports§

pub use adagrad::AdaGrad;
pub use adagrad::AdaGradConfig;
pub use sgd::Sgd;
pub use sgd::SgdConfig;

Modules§

adagrad
AdaGrad optimizer.
sgd
Stochastic gradient descent optimizer.

Enums§

Optimizer
Concrete optimizer enum wrapping all supported optimizers.