SecondOrderOptimizer

Trait SecondOrderOptimizer 

Source
pub trait SecondOrderOptimizer<A: Float + ScalarOperand + Debug + Send + Sync, D: Dimension> {
    // Required methods
    fn step_second_order(
        &mut self,
        params: &Array<A, D>,
        gradients: &Array<A, D>,
        hessian_info: &HessianInfo<A, D>,
    ) -> Result<Array<A, D>>;
    fn reset(&mut self);
}
Expand description

Trait for second-order optimization methods

Required Methods§

Source

fn step_second_order( &mut self, params: &Array<A, D>, gradients: &Array<A, D>, hessian_info: &HessianInfo<A, D>, ) -> Result<Array<A, D>>

Update parameters using second-order information

Source

fn reset(&mut self)

Reset optimizer state

Implementors§