Skip to main content

gradient_descent

Function gradient_descent 

Source
pub fn gradient_descent<T, F, G>(
    initial: &Array1<T>,
    objective: F,
    gradient: G,
    config: &SGDConfig<T>,
) -> Result<Array1<T>, OptimizationError>
where T: NabledReal, F: Fn(&Array1<T>) -> T, G: Fn(&Array1<T>) -> Array1<T>,
Expand description

Minimize objective with fixed-step gradient descent.

ยงErrors

Returns an error for invalid inputs/configuration or non-finite gradients.