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
use num::{Float, FromPrimitive};
pub fn schaffer_n2<T: Float + FromPrimitive>(param: &[T]) -> T {
let plen = param.len();
assert!(plen == 2);
let (x1, x2) = (param[0], param[1]);
let n05 = T::from_f64(0.5).unwrap();
let n1 = T::from_f64(1.0).unwrap();
let n0001 = T::from_f64(0.0001).unwrap();
n05 + ((x1.powi(2) - x2.powi(2)).sin().powi(2) - n05)
/ (n1 + n0001 * (x1.powi(2) + x2.powi(2))).powi(2)
}
pub fn schaffer_n4<T: Float + FromPrimitive>(param: &[T]) -> T {
let plen = param.len();
assert!(plen == 2);
let (x1, x2) = (param[0], param[1]);
let n05 = T::from_f64(0.5).unwrap();
let n1 = T::from_f64(1.0).unwrap();
let n0001 = T::from_f64(0.0001).unwrap();
n05 + ((x1.powi(2) - x2.powi(2)).abs().sin().cos().powi(2) - n05)
/ (n1 + n0001 * (x1.powi(2) + x2.powi(2))).powi(2)
}
mod tests {
#[test]
fn test_schaffer_n2_optimum() {
assert!((::schaffer_n2(&[0_f32, 0_f32])).abs() < ::std::f32::EPSILON);
assert!((::schaffer_n2(&[0_f64, 0_f64])).abs() < ::std::f64::EPSILON);
}
#[test]
fn test_schaffer_n4_optimum() {
assert!((::schaffer_n4(&[0_f32, 1.25313_f32]) - 0.291992).abs() < ::std::f32::EPSILON);
}
#[test]
#[should_panic]
fn test_schaffer_n2_param_length() {
::schaffer_n2(&[0.0_f32, 0.0_f32, 0.0_f32]);
}
#[test]
#[should_panic]
fn test_schaffer_n4_param_length() {
::schaffer_n4(&[0.0_f32, 0.0_f32, 0.0_f32]);
}
}