Manuel Raimann
b36137bc96
Add micro-benchmarks for TPE, Random, and Grid samplers with varying history sizes, end-to-end optimization benchmarks on Sphere and Rosenbrock, and a convergence tracking example that outputs CSV data. Includes 6 standard test functions (Sphere, Rosenbrock, Rastrigin, Ackley, Branin, Hartmann6) with correctness tests at known optima.
46 lines
1.0 KiB
Rust
46 lines
1.0 KiB
Rust
#[path = "../benches/test_functions.rs"]
|
|
mod test_functions;
|
|
|
|
use test_functions::*;
|
|
|
|
const TOL: f64 = 1e-10;
|
|
|
|
#[test]
|
|
fn sphere_at_optimum() {
|
|
assert!(sphere(&[0.0, 0.0]).abs() < TOL);
|
|
assert!(sphere(&[0.0; 10]).abs() < TOL);
|
|
}
|
|
|
|
#[test]
|
|
fn rosenbrock_at_optimum() {
|
|
assert!(rosenbrock(&[1.0, 1.0]).abs() < TOL);
|
|
assert!(rosenbrock(&[1.0; 5]).abs() < TOL);
|
|
}
|
|
|
|
#[test]
|
|
fn rastrigin_at_optimum() {
|
|
assert!(rastrigin(&[0.0, 0.0]).abs() < TOL);
|
|
assert!(rastrigin(&[0.0; 10]).abs() < TOL);
|
|
}
|
|
|
|
#[test]
|
|
fn ackley_at_optimum() {
|
|
assert!(ackley(&[0.0, 0.0]).abs() < 1e-8);
|
|
assert!(ackley(&[0.0; 10]).abs() < 1e-8);
|
|
}
|
|
|
|
#[test]
|
|
fn branin_at_optimum() {
|
|
let target = 0.397_887_357_729_738_1;
|
|
let val = branin(&[std::f64::consts::PI, 2.275]);
|
|
assert!((val - target).abs() < 1e-3);
|
|
}
|
|
|
|
#[test]
|
|
fn hartmann6_at_optimum() {
|
|
let target = -3.3224;
|
|
let x_opt = [0.20169, 0.150011, 0.476874, 0.275332, 0.311652, 0.6573];
|
|
let val = hartmann6(&x_opt);
|
|
assert!((val - target).abs() < 0.01);
|
|
}
|