from tensorflow.keras.callbacks import *
earlystop_callback = EarlyStopping(
monitor='val_precision', min_delta=0.0001,
verbose=1, patience=6)
# patience=6 早停止后再迭代6次
batch_size = 2048
train_nums = len(pdtrain)
deep.fit_generator(
GeneratorRandomPatchs(train_x, train_y, batch_size, train_nums, feature_names),
validation_data=(val_x, val_y),
steps_per_epoch=train_nums // batch_size,
epochs=10,
verbose=2,
shuffle=True,
callbacks=[earlystop_callback]
)
https://www.w3cschool.cn/tensorflow_python/tf_keras_callbacks_EarlyStopping.html