Как можно эффективно остановить процесс подгонки модели обучения с помощью обратного вызова в keras? До сих пор я пробовал различные подходы, в том числе приведенный ниже.
class EarlyStoppingCallback(tf.keras.callbacks.Callback):
def __init__(self, threshold):
super(EarlyStoppingCallback, self).__init__()
self.threshold = threshold
def on_epoch_end(self, epoch, logs=None):
accuracy = logs["accuracy"]
if accuracy >= self.threshold:
print("Stopping early!")
self.model.stop_training = True
Обратный вызов выполняется, однако self.model.stop_training = True, похоже, не действует. Печать выполнена успешно, но модель продолжает обучение. Есть идеи, как решить эту проблему? Моя версия тензорного потока: tenorflow == 1.14.0