def test_get_search_iteration_manager(self): # Grid search experiment_group = ExperimentGroupFactory() assert isinstance(get_search_iteration_manager(experiment_group), BaseIterationManager) # Random search experiment_group = ExperimentGroupFactory( content=experiment_group_spec_content_early_stopping) assert isinstance(get_search_iteration_manager(experiment_group), BaseIterationManager) # Hyperband experiment_group = ExperimentGroupFactory( content=experiment_group_spec_content_hyperband) assert isinstance(get_search_iteration_manager(experiment_group), HyperbandIterationManager) # BO experiment_group = ExperimentGroupFactory( content=experiment_group_spec_content_bo) assert isinstance(get_search_iteration_manager(experiment_group), BOIterationManager)
def test_get_search_iteration_manager(self): # Grid search experiment_group = ExperimentGroupFactory() assert get_search_iteration_manager(experiment_group) is None # Random search experiment_group = ExperimentGroupFactory( content=experiment_group_spec_content_early_stopping) assert get_search_iteration_manager(experiment_group) is None # Hyperband experiment_group = ExperimentGroupFactory( content=experiment_group_spec_content_hyperband) assert isinstance(get_search_iteration_manager(experiment_group), HyperbandIterationManager) # BO experiment_group = ExperimentGroupFactory( content=experiment_group_spec_content_bo) assert isinstance(get_search_iteration_manager(experiment_group), BOIterationManager)
def iteration_manager(self): from hpsearch.iteration_managers import get_search_iteration_manager return get_search_iteration_manager(experiment_group=self)