tf2.0 EarlyStopping早停止实现

在这里插入图片描述

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

https://zhuanlan.zhihu.com/p/186458993

猜你喜欢

转载自blog.csdn.net/qq_42363032/article/details/121360330