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
|
from functools import partial
from itertools import product
import pytest
from skopt import Optimizer, forest_minimize, gbrt_minimize, gp_minimize
from skopt.learning import ExtraTreesRegressor
# dummy_minimize does not support same parameters so
# treated separately
MINIMIZERS = [gp_minimize]
ACQUISITION = ["LCB", "PI", "EI"]
for est, acq in product(["ET", "RF"], ACQUISITION):
MINIMIZERS.append(partial(forest_minimize, base_estimator=est, acq_func=acq))
for acq in ACQUISITION:
MINIMIZERS.append(partial(gbrt_minimize, acq_func=acq))
def test_n_random_starts_Optimizer():
# n_random_starts got renamed in v0.4
et = ExtraTreesRegressor(random_state=2)
with pytest.deprecated_call():
Optimizer([(0, 1.0)], et, n_random_starts=10, acq_optimizer='sampling')
|
