def test_reg_build_fn(self): clf = wrappers.KerasRegressor( build_fn=build_fn_reg, hidden_dim=HIDDEN_DIM, batch_size=BATCH_SIZE, epochs=EPOCHS, ) self.check_sample_weights_work(clf)
def test_regression_build_fn(self): """Tests for errors using KerasRegressor.""" reg = wrappers.KerasRegressor( build_fn=build_fn_reg, hidden_dim=HIDDEN_DIM, batch_size=BATCH_SIZE, epochs=EPOCHS, ) assert_regression_works(reg)
def test_regression_build_fn(self): """Tests for errors using KerasRegressor.""" reg = wrappers.KerasRegressor( build_fn=build_fn_reg, hidden_dim=HIDDEN_DIM, batch_size=BATCH_SIZE, epochs=EPOCHS, ) # create dataset X = np.random.rand(10, 20) with pytest.raises(NotFittedError): reg.predict(X)
def test_regression_class_build_fn(self): """Tests for errors using KerasRegressor implementing __call__.""" class ClassBuildFnReg: def __call__(self, hidden_dim): return build_fn_reg(hidden_dim) reg = wrappers.KerasRegressor( build_fn=ClassBuildFnReg(), hidden_dim=HIDDEN_DIM, batch_size=BATCH_SIZE, epochs=EPOCHS, ) assert_regression_works(reg)