Skip to content
This repository was archived by the owner on Aug 9, 2023. It is now read-only.

Commit f6d5eda

Browse files
committed
Add warning about early stopping not working
1 parent b10e2a5 commit f6d5eda

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

wellcomeml/ml/cnn.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -271,7 +271,9 @@ def fit(self, X, Y=None, embedding_matrix=None, steps_per_epoch=None):
271271
if train_steps_per_epoch == 0:
272272
logger.warning(
273273
"Not enough data for validation. Consider decreasing \
274-
batch_size or validation_split"
274+
batch_size or validation_split. Some features that \
275+
rely on validation metrics like early stopping might \
276+
not work"
275277
)
276278
else:
277279
steps_per_epoch = train_steps_per_epoch

0 commit comments

Comments
 (0)