scirs2-optimize 0.1.0-alpha.3

Optimization module for SciRS2
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

use ndarray::{array, ArrayView1};
use scirs2_optimize::sparse_numdiff::{sparse_hessian, SparseFiniteDiffOptions};
use scirs2_sparse::sparray::SparseArray;

fn sphere(x: &ArrayView1<f64>) -> f64 {
    x.iter().map(|&xi| xi * xi).sum()
}

fn main() {
    let x = array![1.0, 2.0];

    // Test the default method
    println!("Testing default method:");
    let hess = sparse_hessian(sphere, &x.view(), None, None, None).unwrap();

    println!("Hessian computed:");
    let hess_dense = hess.to_array();
    println!("{:?}", hess_dense);

    // Debug: check the raw sparse data
    let data = hess.get_data();
    let indices = hess.get_indices();
    let indptr = hess.get_indptr();

    println!("\nSparse representation:");
    println!("Data: {:?}", data);
    println!("Indices: {:?}", indices);
    println!("Indptr: {:?}", indptr);

    println!("\nShape: {:?}", hess.shape());

    // Test with explicit 3-point method
    println!("\n\nTesting 3-point method explicitly:");
    let options = SparseFiniteDiffOptions {
        method: "3-point".to_string(),
        ..SparseFiniteDiffOptions::default()
    };

    let hess2 = sparse_hessian(sphere, &x.view(), None, None, Some(options)).unwrap();
    let hess2_dense = hess2.to_array();
    println!("{:?}", hess2_dense);
}