def build_functionally_defined_objects_train_curriculum(
num_samples: Optional[int],
num_noise_objects: Optional[int],
language_generator: LanguageGenerator[HighLevelSemanticsSituation,
LinearizedDependencyTree],
) -> Sequence[Phase1InstanceGroup]:
return [
_make_sit_on_curriculum(num_samples, num_noise_objects,
language_generator),
_make_drink_curriculum(num_samples, num_noise_objects,
language_generator),
]
def build_actions_and_generics_curriculum(
num_samples: Optional[int],
num_noise_objects: Optional[int],
language_generator: LanguageGenerator[HighLevelSemanticsSituation,
LinearizedDependencyTree],
) -> Sequence[Phase1InstanceGroup]:
# pylint: disable=unused-argument
return [
_make_eat_curriculum(10, 0, language_generator),
_make_drink_curriculum(10, 0, language_generator),
_make_sit_curriculum(10, 0, language_generator),
_make_jump_curriculum(10, 0, language_generator),
_make_fly_curriculum(10, 0, language_generator),
_make_generic_statements_curriculum(
num_samples=20,
noise_objects=0,
language_generator=language_generator),
]
"kinds-and-generics": [
_make_each_object_by_itself_curriculum(num_samples, 0,
language_generator),
_make_kind_predicates_curriculum(None, None,
language_generator),
_make_generic_statements_curriculum(
num_samples=3,
noise_objects=0,
language_generator=language_generator),
],
"obj-actions-kinds-generics": [
_make_each_object_by_itself_curriculum(num_samples, 0,
language_generator),
# Actions - verbs in generics
_make_eat_curriculum(10, 0, language_generator),
_make_drink_curriculum(10, 0, language_generator),
_make_sit_curriculum(10, 0, language_generator),
_make_jump_curriculum(10, 0, language_generator),
_make_fly_curriculum(10, 0, language_generator),
# Plurals
_make_plural_objects_curriculum(None, 0, language_generator),
# Color attributes
_make_objects_with_colors_curriculum(None, None,
language_generator),
# Predicates
_make_colour_predicates_curriculum(None, None,
language_generator),
_make_kind_predicates_curriculum(None, None,
language_generator),
# Generics
_make_generic_statements_curriculum(
}
for condition, banned_objects in condition_and_banned_objects.items():
pretraining_curricula = [
_make_each_object_by_itself_curriculum(
num_samples,
0,
language_generator,
banned_ontology_types=banned_objects,
),
# Actions - verbs in generics
_make_eat_curriculum(10,
0,
language_generator,
banned_ontology_types=banned_objects),
_make_drink_curriculum(10,
0,
language_generator,
banned_ontology_types=banned_objects),
_make_sit_curriculum(10,
0,
language_generator,
banned_ontology_types=banned_objects),
_make_jump_curriculum(10,
0,
language_generator,
banned_ontology_types=banned_objects),
_make_fly_curriculum(10,
0,
language_generator,
banned_ontology_types=banned_objects),
# Color attributes
_make_objects_with_colors_curriculum(
def test_drink_curriculum(language_generator):
curriculum_test(_make_drink_curriculum(None, None, language_generator))