Completed
Pull Request — master (#21)
by Grega
01:03
created

Rosenbrock.evaluate()   A

Complexity

Conditions 2

Size

Total Lines 9

Duplication

Lines 0
Ratio 0 %

Importance

Changes 2
Bugs 0 Features 2
Metric Value
dl 0
loc 9
rs 9.6666
c 2
b 0
f 2
cc 2
1
"""Implementation of Rosenbrock benchmark function."""
2
3
import math
4
5
__all__ = ['Rosenbrock']
6
7
8
class Rosenbrock(object):
9
10
    def __init__(self, Lower=-100, Upper=100):
11
        self.Lower = Lower
12
        self.Upper = Upper
13
14
    @classmethod
15
    def function(cls):
16
        def evaluate(D, sol):
17
            val = 0.0
18
19
            for i in range(D - 1):
20
                val = val + 100 * \
21
                    math.pow(sol[i + 1] - math.pow((sol[i]), 2),
22
                             2) + math.pow((sol[i] - 1), 2)
23
24
            return val
25
26
        return evaluate
27