提交 541bef56 编写于 作者: O openeuler-ci-bot 提交者: Gitee

!165 optimizer: don't import TPEOptimizer and KnobSamplingManager when startup

Merge pull request !165 from 谢志鹏/master
......@@ -29,8 +29,6 @@ from skopt.utils import normalize_dimensions
from skopt.utils import cook_estimator
from analysis.optimizer.abtest_tuning_manager import ABtestTuningManager
from analysis.optimizer.knob_sampling_manager import KnobSamplingManager
from analysis.optimizer.tpe_optimizer import TPEOptimizer
from analysis.optimizer.weighted_ensemble_feature_selector import WeightedEnsembleFeatureSelector
LOGGER = logging.getLogger(__name__)
......@@ -329,6 +327,7 @@ class Optimizer(multiprocessing.Process):
# convert string option into index
options = abtuning_manager.get_options_index(options)
elif self.engine == 'lhs':
from analysis.optimizer.knob_sampling_manager import KnobSamplingManager
knobsampling_manager = KnobSamplingManager(self.knobs, self.child_conn,
self.max_eval, self.split_count)
options = knobsampling_manager.get_knob_samples()
......@@ -336,6 +335,7 @@ class Optimizer(multiprocessing.Process):
params = knobsampling_manager.get_best_params(options, performance)
options = knobsampling_manager.get_options_index(options)
elif self.engine == 'tpe':
from analysis.optimizer.tpe_optimizer import TPEOptimizer
tpe_opt = TPEOptimizer(self.knobs, self.child_conn, self.max_eval)
best_params = tpe_opt.tpe_minimize_tuning()
final_param = {}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册