def get_search_iteration_manager(experiment_group):
if SearchAlgorithms.is_hyperband(experiment_group.search_algorithm):
return HyperbandIterationManager(experiment_group=experiment_group)
if SearchAlgorithms.is_bo(experiment_group.search_algorithm):
return BOIterationManager(experiment_group=experiment_group)
return None
def get_iteration_config(search_algorithm, iteration=None):
if SearchAlgorithms.is_hyperband(search_algorithm):
if not iteration:
raise ValueError('No iteration was provided')
return HyperbandIterationConfig.from_dict(iteration)
if SearchAlgorithms.is_bo(search_algorithm):
if not iteration:
raise ValueError('No iteration was provided')
return BOIterationConfig.from_dict(iteration)
return None
def create(experiment_group):
if SearchAlgorithms.is_grid(experiment_group.search_algorithm):
return grid.create(experiment_group=experiment_group)
elif SearchAlgorithms.is_random(experiment_group.search_algorithm):
return random.create(experiment_group=experiment_group)
elif SearchAlgorithms.is_hyperband(experiment_group.search_algorithm):
return hyperband.create(experiment_group=experiment_group)
elif SearchAlgorithms.is_bo(experiment_group.search_algorithm):
return bo.create(experiment_group=experiment_group)
return None
def get_search_algorithm_manager(hptuning_config):
if not hptuning_config:
return None
if SearchAlgorithms.is_grid(hptuning_config.search_algorithm):
return GridSearchManager(hptuning_config=hptuning_config)
if SearchAlgorithms.is_random(hptuning_config.search_algorithm):
return RandomSearchManager(hptuning_config=hptuning_config)
if SearchAlgorithms.is_hyperband(hptuning_config.search_algorithm):
return HyperbandSearchManager(hptuning_config=hptuning_config)
if SearchAlgorithms.is_bo(hptuning_config.search_algorithm):
return BOSearchManager(hptuning_config=hptuning_config)
return None
def get_search_algorithm_manager(params_config):
if not params_config:
return None
if SearchAlgorithms.is_grid(params_config.search_algorithm):
return GridSearchManager(params_config=params_config)
if SearchAlgorithms.is_random(params_config.search_algorithm):
return RandomSearchManager(params_config=params_config)
if SearchAlgorithms.is_hyperband(params_config.search_algorithm):
return HyperbandSearchManager(params_config=params_config)
if SearchAlgorithms.is_bo(params_config.search_algorithm):
return BOSearchManager(params_config=params_config)
return None
def create(experiment_group):
if SearchAlgorithms.is_grid(experiment_group.search_algorithm):
auditor.record(event_type=EXPERIMENT_GROUP_GRID,
instance=experiment_group)
return grid.create(experiment_group=experiment_group)
elif SearchAlgorithms.is_random(experiment_group.search_algorithm):
auditor.record(event_type=EXPERIMENT_GROUP_RANDOM,
instance=experiment_group)
return random.create(experiment_group=experiment_group)
elif SearchAlgorithms.is_hyperband(experiment_group.search_algorithm):
auditor.record(event_type=EXPERIMENT_GROUP_HYPERBAND,
instance=experiment_group)
return hyperband.create(experiment_group=experiment_group)
elif SearchAlgorithms.is_bo(experiment_group.search_algorithm):
auditor.record(event_type=EXPERIMENT_GROUP_BO,
instance=experiment_group)
return bo.create(experiment_group=experiment_group)
return None
def create(experiment_group):
if SearchAlgorithms.is_grid(experiment_group.search_algorithm):
auditor.record(event_type=EXPERIMENT_GROUP_GRID,
instance=experiment_group)
return grid.create(experiment_group=experiment_group)
elif SearchAlgorithms.is_random(experiment_group.search_algorithm):
auditor.record(event_type=EXPERIMENT_GROUP_RANDOM,
instance=experiment_group)
return random.create(experiment_group=experiment_group)
elif SearchAlgorithms.is_hyperband(experiment_group.search_algorithm):
auditor.record(event_type=EXPERIMENT_GROUP_HYPERBAND,
instance=experiment_group)
return hyperband.create(experiment_group=experiment_group)
elif SearchAlgorithms.is_bo(experiment_group.search_algorithm):
auditor.record(event_type=EXPERIMENT_GROUP_BO,
instance=experiment_group)
return bo.create(experiment_group=experiment_group)
return None