def rung_2(rung_1):
"""Create fake points and objectives for rung 1."""
points = [
trial.params["lr"]
for _, trial in sorted(rung_1["results"].values())[:1]
]
return create_rung_from_points(points, n_trials=1, resources=9)
def rung_1(rung_0: RungDict):
"""Create fake points and objectives for rung 1."""
points = [
trial.params["lr"]
for _, trial in sorted(rung_0["results"].values())[:3]
]
return create_rung_from_points(points, n_trials=3, resources=3)
def big_rung_0():
"""Create fake points and objectives for big rung 0."""
n_rung_0 = 9 * 3
return create_rung_from_points(
np.linspace(0, n_rung_0 - 1, n_rung_0),
n_trials=n_rung_0 * 2,
resources=1,
)
def big_rung_1(big_rung_0):
"""Create fake points and objectives for big rung 1."""
n_rung_0 = len(big_rung_0["results"])
n_rung_1 = 3 * 2
return create_rung_from_points(
np.linspace(n_rung_0 - n_rung_1, n_rung_0 - 1, n_rung_1),
n_trials=n_rung_1 * 2,
resources=3,
)
def rung_0():
"""Create fake points and objectives for rung 0."""
return create_rung_from_points(np.linspace(0, 8, 9),
n_trials=9,
resources=1)