Completed
Push — master ( ce1e03...f67568 )
by Simon
14:21
created

DecisionTreeOptimizer.__init__()   A

Complexity

Conditions 1

Size

Total Lines 4
Code Lines 4

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
eloc 4
nop 2
dl 0
loc 4
rs 10
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
8
9
class DecisionTreeOptimizer(BayesianOptimizer):
10
    """Based on the forest-optimizer in the scikit-optimize package"""
11
12
    def __init__(self, _opt_args_):
13
        super().__init__(_opt_args_)
14
        self.n_positioners = 1
15
        self.regr = _opt_args_.tree_regressor
16