def test_CNN_hyperparameters_fcnodes(self):
""" Number of fc nodes from range [123, 123] should be 123. """
custom_settings = self.get_default()
kwargs = {'cnn_min_fc_nodes': 123, 'cnn_max_fc_nodes': 123}
# Replace default parameters with input
for key, value in kwargs.items():
if key in custom_settings:
custom_settings[key] = value
hyperparams = modelgen.generate_CNN_hyperparameter_set(custom_settings)
assert hyperparams.get('fc_hidden_nodes') == 123
def test_CNN_hyperparameters_nrlayers(self):
""" Number of Conv layers from range [4, 4] should be 4. """
custom_settings = self.get_default()
kwargs = {'cnn_min_layers': 4, 'cnn_max_layers': 4}
# Replace default parameters with input
for key, value in kwargs.items():
if key in custom_settings:
custom_settings[key] = value
hyperparams = modelgen.generate_CNN_hyperparameter_set(custom_settings)
assert len(hyperparams.get('filters')) == 4
def test_CNN_hyperparameters_nrlayers(self):
""" Number of fc nodes from range [123, 123] should be 123. """
hyperparams = modelgen.generate_CNN_hyperparameter_set(
min_fc_nodes=123, max_fc_nodes=123)
assert_equal(hyperparams.get('fc_hidden_nodes'), 123)
def test_CNN_hyperparameters_nrlayers(self):
""" Number of Conv layers from range [4, 4] should be 4. """
hyperparams = modelgen.generate_CNN_hyperparameter_set(min_layers=4,
max_layers=4)
assert_equal(len(hyperparams.get('filters')), 4)
