Skip to content

Commit cd735f9

Browse files
add test high dataset predict
1 parent 489c4e8 commit cd735f9

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/test_utils.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -265,6 +265,13 @@ def test_check_network_compile():
265265
"Please use `model.compile(optimizer, loss)`."
266266
in str(excinfo.value))
267267

268+
269+
def test_check_network_high_dataset():
270+
Xs, ys, Xt, yt = make_regression_da(100000, 1001)
271+
net = _get_model_Sequential(compiled=True)
272+
new_net = check_network(net, copy=True, compile_=True)
273+
new_net.predict(Xs)
274+
268275

269276
estimators = [
270277
Ridge(),

0 commit comments

Comments
 (0)