def test_hyperparameters_config(): """Checks that the model has the right hyperparameter configuration.""" # Clear the session to restart the naming of the Keras layers K.clear_session() m = Member(get_test_model, steps_to_ready, tune_lr=True) config = m.get_hyperparameter_config() expected_config = { 'l1:0': 0.1, 'l2:0': 1e-5, 'dr:1': 0.1, 'l1:2': 0.2, 'l2:2': 1e-6, 'lr': 0.1 } for k, v in expected_config.items(): assert v == pytest.approx(config[k], abs=1e-6)
def test_regularizers_found(): """Checks that the member could find the two regularizers added.""" m = Member(get_test_model, steps_to_ready) assert len(m.hyperparameters) == 3
def test_learning_rate_found(): """Checks that the member knows that it should tune the learning rate.""" m = Member(get_test_model, steps_to_ready, tune_lr=True) assert len(m.hyperparameters) == 4
def get_test_member(): return Member(get_test_model, steps_to_ready, tune_lr=True)