optirs-core 0.3.2

OptiRS core optimization algorithms and utilities
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
use optirs_core::optimizers::{Adam, Optimizer};
use scirs2_core::ndarray::array;

#[allow(dead_code)]
fn main() {
    let mut adam = Adam::new(0.001);
    adam.set_beta1(0.9);
    adam.set_beta2(0.999);

    let params = array![1.0, 2.0, 3.0];
    let gradients = array![0.1, 0.2, 0.3];

    match adam.step(&params, &gradients) {
        Ok(updated) => println!("Updated params: {}", updated),
        Err(e) => eprintln!("Error: {}", e),
    }
}