def get_bo_tuner(
matrix: V1Bayes,
configs: List[Dict],
metrics: List[float],
iteration: int,
container: V1Container = None,
) -> V1Operation:
container = container or get_default_tuner_container(["polyaxon", "tuner", "bo"])
return get_tuner(
name="bayesian-tuner",
container=container,
matrix=matrix,
configs=configs,
metrics=metrics,
iteration=iteration,
)
def get_hyperopt_tuner(
matrix: V1Hyperopt,
join: V1Join,
iteration: int,
) -> V1Operation:
iteration = matrix.create_iteration(iteration)
return get_tuner(
name="hyperopt-tuner",
container=get_container(
tuner_container=get_default_tuner_container(
["polyaxon", "tuner", "hyperopt"]),
container=matrix.container,
),
matrix=matrix,
join=join,
iteration=iteration,
)
def get_bo_tuner(
matrix: V1Bayes,
join: V1Join,
iteration: int,
) -> V1Operation:
iteration = matrix.create_iteration(iteration)
return get_tuner(
name="bayesian-tuner",
container=get_container(
tuner_container=get_default_tuner_container(
["polyaxon", "tuner", "bayes"]),
container=matrix.container,
),
matrix=matrix,
join=join,
iteration=iteration,
)
def get_hyperband_tuner(
matrix: V1Hyperband,
configs: List[Dict],
metrics: List[float],
iteration: int,
container: V1Container = None,
) -> V1Operation:
container = container or get_default_tuner_container(
["polyaxon", "tuner", "hyperband"]
)
return get_tuner(
name="hyperband-tuner",
container=container,
matrix=matrix,
configs=configs,
metrics=metrics,
iteration=iteration,
)
def get_hyperband_tuner(
matrix: V1Hyperband,
join: V1Join,
iteration: int,
bracket_iteration: int,
) -> V1Operation:
matrix.set_tuning_params()
iteration, bracket_iteration = matrix.create_iteration(
iteration, bracket_iteration)
return get_tuner(
name="hyperband-tuner",
container=get_container(
tuner_container=get_default_tuner_container(
["polyaxon", "tuner", "hyperband"],
bracket_iteration=bracket_iteration,
),
container=matrix.container,
),
matrix=matrix,
join=join,
iteration=iteration,
bracket_iteration=bracket_iteration,
)