Total Complexity | 1 |
Total Lines | 16 |
Duplicated Lines | 0 % |
Changes | 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 |