basin 1.5.0

Numerical optimization in pure Rust, with pluggable linear-algebra backends and WASM support.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

#![cfg(feature = "ndarray")]

//! ndarray-backend mirror of the `Vec<f64>` Solis-Wets suite
//! (`tests/solis_wets_vec.rs`): reduced to the checks that exercise the
//! per-backend trait wiring.

use basin::problems::Sphere;
use basin::{Executor, RhoTolerance, SolisWets, TerminationReason};
use ndarray::Array1;

#[test]
fn same_seed_yields_identical_trajectory() {
    let run = || {
        Executor::from_start(
            Sphere::<Array1<f64>>::new(),
            SolisWets::new(42),
            Array1::from(vec![2.0, -1.5]),
        )
        .max_iter(200)
        .run()
        .unwrap()
    };
    let result_a = run();
    let result_b = run();
    assert_eq!(result_a.cost(), result_b.cost());
    assert_eq!(result_a.param(), result_b.param());
}

#[test]
fn converges_on_sphere_5d_via_rho_tolerance() {
    let result = Executor::from_start(
        Sphere::<Array1<f64>>::new(),
        SolisWets::new(7),
        Array1::from(vec![2.0, -1.0, 1.5, 0.5, -2.0]),
    )
    .terminate_on(RhoTolerance::new(1e-8))
    .max_iter(100_000)
    .run()
    .unwrap();

    assert_eq!(result.reason, TerminationReason::RhoTolerance);
    assert!(
        result.cost() < 1e-6,
        "sphere 5-D cost = {} (expected < 1e-6)",
        result.cost()
    );
}