Crate vikos [−] [src]
A machine learning library for supervised regression trainings
This library wants to enable its user to write training algorithms
independent of the model trained or the cost function tried to
minimize.
Consequently its two main traits are currently Model
and Cost
.
The two submodules model
and cost
provide ready to use
implementations of said traits.
Modules
cost |
Implementations of |
linear_algebra |
Defines linear algebra traits used for some model parameters |
model |
Implementations of |
Traits
Cost |
Cost functions those value is supposed be minimized by the training algorithm |
Model |
A Model is defines how to predict a target from an input |
Functions
gradient_descent_step |
An SGD training step with a velocity term |
inert_gradient_descent_step |
Changes all coefficents of model based on their derivation of the cost function at features |
inert_stochastic_gradient_descent |
SGD tranining with constant learning rate and velocity |
stochastic_gradient_descent |
Applies a plain SGD training step to model once for every event in history using a constant learning rate |