def get_curriculum_four(
epochs, num_nodes, train_size, val_size, seed, regen, debug
):
"""Create curriculum for experiment four."""
steps = [
(get_horiz_param, 2),
(get_vert_param, 2),
(get_plus_param, 2),
(get_down_line_param, 2),
(get_up_line_param, 2),
(get_x_shape_param, 2),
(get_uniform_param, epochs - 12),
]
val_param = get_uniform_param(NUM_TILES)
return make_curriculum(
epochs,
num_nodes,
train_size,
val_size,
steps,
val_param,
seed,
regen,
debug,
)
def get_curriculum_one(
epochs, num_nodes, train_size, val_size, seed, regen, debug
):
"""Create curriculum for experiment one.
Curriculum consists of:
1) Tiny-pair 6
2) Medium-pair 6
3) Uniform 8
"""
steps = [
(get_tiny_pair_param, 6),
(get_medium_pair_param, 6),
(get_uniform_param, epochs - 12),
]
val_param = get_uniform_param(NUM_TILES)
return make_curriculum(
epochs,
num_nodes,
train_size,
val_size,
steps,
val_param,
seed,
regen,
debug,
)
def get_curriculum_two(
epochs, num_nodes, train_size, val_size, seed, regen, debug
):
"""Create curriculum for experiment two.
Curriculum consists of:
1) Tiny-pair 4
2) Tiny-quad 4
3) Border 4
4) Uniform 8
"""
steps = [
(get_tiny_pair_param, 4),
(get_tiny_quad_param, 4),
(get_border_param, 4),
(get_uniform_param, epochs - 12),
]
val_param = get_uniform_param(NUM_TILES)
return make_curriculum(
epochs,
num_nodes,
train_size,
val_size,
steps,
val_param,
seed,
regen,
debug,
)
def get_curriculum_zero(
epochs, num_nodes, train_size, val_size, seed, regen, debug
):
"""Create the uniform curriculum."""
steps = [(get_uniform_param, epochs)]
val_param = get_uniform_param(NUM_TILES)
return make_curriculum(
epochs,
num_nodes,
train_size,
val_size,
steps,
val_param,
seed,
regen,
debug,
)
def get_curriculum_eighteen(
epochs, num_nodes, train_size, val_size, seed, regen, debug
):
"""Create curriculum for experiment seventeen."""
steps = [
(get_circle_param, 10),
(get_uniform_param, 20),
]
val_param = get_uniform_param(NUM_TILES)
return make_curriculum(
epochs,
num_nodes,
train_size,
val_size,
steps,
val_param,
seed,
regen,
debug,
)
def get_curriculum_nine(
epochs, num_nodes, train_size, val_size, seed, regen, debug
):
"""Create curriculum for experiment nine."""
steps = [
(get_down_line_param, 10),
(get_uniform_param, epochs - 10),
]
val_param = get_uniform_param(NUM_TILES)
return make_curriculum(
epochs,
num_nodes,
train_size,
val_size,
steps,
val_param,
seed,
regen,
debug,
)
def get_curriculum_three(
epochs, num_nodes, train_size, val_size, seed, regen, debug
):
"""Create curriculum for experiment three."""
steps = [
(get_tiny_pair_param, 4),
(get_tiny_quad_param, 4),
(get_circle_param, 4),
(get_uniform_param, epochs - 12),
]
val_param = get_uniform_param(NUM_TILES)
return make_curriculum(
epochs,
num_nodes,
train_size,
val_size,
steps,
val_param,
seed,
regen,
debug,
)