tensorflow通过callback提前停止训练

tf提前停止训练以防达到过拟合

以下方法基于类来实现

class Mycallback(tf.keras.callbacks.Callback):
    def on_epoch_end(self,epoch,acc=0.22,logs={}):
        if logs.get("loss")<=acc:
            print("stop")
            self.model.stop_training=True
callback=Mycallback()

以下是运行结果

其他方法参考其他博客