def test_pickModel(self):
model_library = ModelLibrary('classification', 'small')
available_model_names = list(model_library.model_libraries['small classification'])
model_name = model_library.pickModel(model_selection_method='random')
self.assertTrue(model_name in available_model_names)
def test_pickParameters(self):
model_library = ModelLibrary('classification', 'small')
model_name = list(model_library.model_libraries['small classification'])[0]
uninitialized_model = model_library.model_libraries['small classification'][model_name]
ModelClass, parameter_ranges = uninitialized_model['model'], uninitialized_model['parameter_ranges']
chosen_parameters = model_library.pickParameters(parameter_ranges, parameter_selection_method='random')
for param in parameter_ranges:
self.assertTrue(chosen_parameters[param] in parameter_ranges[param])
def test_generateModel(self):
model_library = ModelLibrary('classification', 'small')
model = model_library.generateModel(model_selection_method='random', parameter_selection_method='random')
