autoeq-de 0.2.52

Non linear optimisation library with own DE solvers and interface to NLOpt and MetaHeuristics
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

//! Tests for metadata-driven optimization examples

#[cfg(test)]
mod tests {
    use crate::{
        DEConfig, DEConfigBuilder, Mutation, Strategy, differential_evolution,
        run_recorded_differential_evolution,
    };
    use autoeq_testfunctions::{get_function_bounds_2d, get_function_bounds_vec, rosenbrock};

    /// Example test showing how to use metadata for bounds
    #[test]
    fn test_de_rosenbrock_with_metadata_bounds() {
        // Use metadata bounds instead of hardcoded ones
        let bounds = get_function_bounds_2d("rosenbrock", (-5.0, 5.0)); // fallback if not found

        let mut config = DEConfig::default();
        config.seed = Some(42);
        config.maxiter = 500;
        config.popsize = 40;
        config.recombination = 0.9;
        config.strategy = Strategy::RandToBest1Exp;

        let result = differential_evolution(&rosenbrock, &bounds, config);

        // Rosenbrock function: Global minimum f(x) = 0 at x = (1, 1)
        assert!(result.fun < 1e-3);

        // Check that solution is close to expected optimum
        assert!(
            (result.x[0] - 1.0).abs() < 0.1,
            "x[0] should be close to 1.0: {}",
            result.x[0]
        );
        assert!(
            (result.x[1] - 1.0).abs() < 0.1,
            "x[1] should be close to 1.0: {}",
            result.x[1]
        );

        println!("Used bounds: {:?}", bounds);
        println!(
            "Found solution: ({:.4}, {:.4}) with f = {:.6}",
            result.x[0], result.x[1], result.fun
        );
    }

    #[test]
    fn test_de_rosenbrock_recorded_with_metadata() {
        // Use metadata bounds for recorded optimization
        let bounds = get_function_bounds_vec("rosenbrock", (-5.0, 5.0));
        let config = DEConfigBuilder::new()
            .seed(123)
            .maxiter(500)
            .popsize(40)
            .strategy(Strategy::RandToBest1Exp)
            .mutation(Mutation::Factor(0.7))
            .recombination(0.9)
            .build();

        // Run the recorded optimization (requires AUTOEQ_DIR to be set)
        let result =
            run_recorded_differential_evolution("rosenbrock_metadata", rosenbrock, &bounds, config);

        match result {
            Ok((report, _csv_path)) => {
                assert!(report.fun < 1e-3);

                // Check that solution is close to expected optimum (1, 1)
                assert!(
                    report.x[0] > 0.9 && report.x[0] < 1.1,
                    "x[0] not close to 1: {}",
                    report.x[0]
                );
                assert!(
                    report.x[1] > 0.9 && report.x[1] < 1.1,
                    "x[1] not close to 1: {}",
                    report.x[1]
                );

                println!("Used bounds: {:?}", bounds);
                println!(
                    "Found solution: ({:.4}, {:.4}) with f = {:.6}",
                    report.x[0], report.x[1], report.fun
                );
            }
            Err(e) => {
                panic!(
                    "Test requires AUTOEQ_DIR to be set. Error: {}\nPlease run: export AUTOEQ_DIR=/path/to/autoeq",
                    e
                );
            }
        }
    }
}