-
Notifications
You must be signed in to change notification settings - Fork 0
/
benchmark.py
65 lines (52 loc) · 1.45 KB
/
benchmark.py
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
"""
Module to store benchmark functions
"""
import numpy as np
def griewank_2d(x, y, x0=50, y0=50, level=100):
"""
compute the 2D Griewank Function (upside down by level)
:param x: float of x
:param y: float of y
:param x0: float of x center
:param y0: float of y center
:param level: level to upside down
:return: float of Griewank Function
"""
# set center
x = x - x0
y = y - y0
return level - 100 * (((np.square(x) + np.square(y)) / 4000) - (np.cos(x) * np.cos(y / np.sqrt(2))) + 1)
def paraboloid_2d(x, y, x0=50, y0=50, level=100):
"""
(upside down by level)
:param x:
:param y:
:param x0:
:param y0:
:param level:
:return:
"""
# set center
x = x - x0
y = y - y0
return level - (np.square(x) + np.square(y))
def rastrigin_2d(x, y, x0=50, y0=50, level=100):
# set center
x = x - x0
y = y - y0
return level - (20 +
(np.square(x) - 10 * np.cos(2 * np.pi * x))
+ (np.square(y) - 10 * np.cos(2 * np.pi * y)))
def himmelblaus(x, y, x0=50, y0=50, level=1000):
# set center
x = x - x0
y = y - y0
return level - (np.square(np.square(x) + y - 11) + np.square(x + np.square(y) - 7))
def rastrigin(xs, xc=50, level=100):
n = len(xs)
# set center:
x = xs - xc
aux = (10 * n)
for i in range(n):
aux = aux + (np.square(x[i]) - 10 * np.cos(2 * np.pi * x[i]))
return level - aux