def __init__(self, manipulator, extra_seeds=None, **kwargs):
super(SearchDriver, self).__init__(**kwargs)
if extra_seeds is None:
extra_seeds = []
self.manipulator = manipulator
self.wait_for_results = self.tuning_run_main.results_wait
self.commit = self.tuning_run_main.commit
self.generation = 0
self.test_count = 0
self.plugins = plugin.get_enabled(self.args)
self.pending_result_callbacks = list(
) # (DesiredResult, function) tuples
# deepcopy is required to have multiple tuning runs in a single process
if self.args.list_techniques:
techniques, generators = technique.all_techniques()
for t in techniques:
print t.name
sys.exit(0)
if self.args.generate_bandit_technique:
# generate a bandit
self.root_technique = AUCBanditMetaTechnique.generate_technique(
manipulator)
else:
self.root_technique = copy.deepcopy(technique.get_root(self.args))
if isinstance(self.root_technique, AUCBanditMetaTechnique):
self.session.flush()
info = BanditInfo(
tuning_run=self.tuning_run,
c=self.root_technique.bandit.C,
window=self.root_technique.bandit.window,
)
self.session.add(info)
for t in self.root_technique.techniques:
subtechnique = BanditSubTechnique(bandit_info=info,
name=t.name)
self.session.add(subtechnique)
self.objective.set_driver(self)
self.pending_config_ids = set()
self.best_result = None
for t in self.plugins:
t.set_driver(self)
self.root_technique.set_driver(self)
self.seed_cfgs = list(extra_seeds)
for cfg_filename in reversed(self.args.seed_configuration):
if os.path.exists(cfg_filename):
self.seed_cfgs.append(manipulator.load_from_file(cfg_filename))
else:
log.error('no such file for --seed-configuration %s',
cfg_filename)
self.plugins.sort(key=_.priority)
def __init__(self, manipulator, extra_seeds=None, extra_criteria=None, **kwargs):
super(SearchDriver, self).__init__(**kwargs)
if extra_seeds is None:
extra_seeds = []
self.manipulator = manipulator
self.wait_for_results = self.tuning_run_main.results_wait
self.commit = self.tuning_run_main.commit
self.extra_criteria = extra_criteria
self.generation = 0
self.test_count = 0
self.plugins = plugin.get_enabled(self.args)
self.pending_result_callbacks = list() # (DesiredResult, function) tuples
# deepcopy is required to have multiple tuning runs in a single process
if self.args.list_techniques:
techniques, generators = technique.all_techniques()
for t in techniques:
print(t.name)
sys.exit(0)
if self.args.generate_bandit_technique:
# generate a bandit
self.root_technique = AUCBanditMetaTechnique.generate_technique(manipulator)
else:
self.root_technique = copy.deepcopy(technique.get_root(self.args))
if isinstance(self.root_technique, AUCBanditMetaTechnique):
self.session.flush()
info = BanditInfo(tuning_run=self.tuning_run,
c=self.root_technique.bandit.C,
window=self.root_technique.bandit.window,)
self.session.add(info)
for t in self.root_technique.techniques:
subtechnique = BanditSubTechnique(bandit_info=info,
name=t.name)
self.session.add(subtechnique)
self.objective.set_driver(self)
self.pending_config_ids = set()
self.best_result = None
self.new_results = []
for t in self.plugins:
t.set_driver(self)
self.root_technique.set_driver(self)
self.seed_cfgs = list(extra_seeds)
for cfg_filename in reversed(self.args.seed_configuration):
if os.path.exists(cfg_filename):
self.seed_cfgs.append(manipulator.load_from_file(cfg_filename))
else:
log.error('no such file for --seed-configuration %s', cfg_filename)
self.plugins.sort(key=_.priority)