Passed
Push — master ( 2d5f27...fd0b7f )
by Simon
01:30
created

tests.TPE.get_score()   A

Complexity

Conditions 1

Size

Total Lines 2
Code Lines 2

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
eloc 2
dl 0
loc 2
rs 10
c 0
b 0
f 0
cc 1
nop 1
1
# Author: Simon Blanke
2
# Email: [email protected]
3
# License: MIT License
4
5
import numpy as np
6
7
from gradient_free_optimizers import TreeStructuredParzenEstimators
8
9
n_iter = 100
10
11
12
def get_score(pos_new):
13
    return -(pos_new[0] * pos_new[0])
14
15
16
space_dim = np.array([10])
17
init_positions = [np.array([0]), np.array([1]), np.array([2]), np.array([3])]
18
19
20 View Code Duplication
def _base_test(opt, init_positions):
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
21
    for nth_init in range(len(init_positions)):
22
        pos_new = opt.init_pos(nth_init)
23
        score_new = get_score(pos_new)
24
        opt.evaluate(score_new)
25
26
    for nth_iter in range(len(init_positions), n_iter):
27
        pos_new = opt.iterate(nth_iter)
28
        score_new = get_score(pos_new)
29
        opt.evaluate(score_new)
30
31
32
def _test_TreeStructuredParzenEstimators(
33
    init_positions=init_positions, space_dim=space_dim, opt_para={}
34
):
35
    opt = TreeStructuredParzenEstimators(init_positions, space_dim, opt_para)
36
    _base_test(opt, init_positions)
37
38
39
def test_start_up_evals():
40
    for start_up_evals in [1, 100]:
41
        opt_para = {"start_up_evals": start_up_evals}
42
        _test_TreeStructuredParzenEstimators(opt_para=opt_para)
43
44
45
def test_warm_start_smbo():
46
    for warm_start_smbo in [True, False]:
47
        opt_para = {"warm_start_smbo": warm_start_smbo}
48
        _test_TreeStructuredParzenEstimators(opt_para=opt_para)
49