Passed
Push — master ( 300275...433e71 )
by Simon
01:13
created

EnsembleOptimizer.__init__()   A

Complexity

Conditions 1

Size

Total Lines 12
Code Lines 10

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
eloc 10
nop 3
dl 0
loc 12
rs 9.9
c 0
b 0
f 0
1
# Author: Simon Blanke
2
# Email: [email protected]
3
# License: MIT License
4
5
6
from .bayesian_optimization import BayesianOptimizer
7
from .surrogate_models import EnsembleRegressor
8
9
10
from sklearn.neighbors import KNeighborsRegressor
11
from sklearn.neural_network import MLPRegressor
12
from sklearn.tree import DecisionTreeRegressor
13
from sklearn.ensemble import GradientBoostingRegressor
14
from sklearn.svm import SVR
15
from sklearn.gaussian_process import GaussianProcessRegressor
16
17
18
class EnsembleOptimizer(BayesianOptimizer):
19
    def __init__(
20
        self,
21
        search_space,
22
        estimators=[
23
            GradientBoostingRegressor(n_estimators=10),
24
            SVR(),
25
            DecisionTreeRegressor(),
26
            GaussianProcessRegressor(),
27
        ],
28
    ):
29
        super().__init__(search_space)
30
        self.regr = EnsembleRegressor(estimators)
31