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)