scirs2_neural::optimizers

Trait Optimizer

Source 
pub trait Optimizer<F: Float + Debug + ScalarOperand> {
    // Required methods
    fn update(
        &mut self,
        params: &mut [Array<F, IxDyn>],
        grads: &[Array<F, IxDyn>],
    ) -> Result<()>;
    fn get_learning_rate(&self) -> F;
    fn set_learning_rate(&mut self, lr: F);
}
Expand description

Trait for neural network optimizers

Required Methods§

Source

fn update( &mut self, params: &mut [Array<F, IxDyn>], grads: &[Array<F, IxDyn>], ) -> Result<()>

Update parameters based on gradients

Source

fn get_learning_rate(&self) -> F

Get the current learning rate

Source

fn set_learning_rate(&mut self, lr: F)

Set the learning rate

Implementors§