Wie kann ich Keras mitteilen, dass das Training basierend auf dem Verlustwert beendet wird?
Zur Zeit benutze ich den folgenden Code:
callbacks = [
EarlyStopping(monitor='val_loss', patience=2, verbose=0),
ModelCheckpoint(kfold_weights_path, monitor='val_loss', save_best_only=True, verbose=0),
]
model.fit(X_train.astype('float32'), Y_train, batch_size=batch_size, nb_epoch=nb_epoch,
shuffle=True, verbose=1, validation_data=(X_valid, Y_valid),
callbacks=callbacks)
Es weist Keras an, das Training abzubrechen, wenn sich der Verlust 2 Epochen lang nicht verbessert hat. Aber ich möchte aufhören zu trainieren, nachdem der Verlust kleiner als ein konstanter "THR" wurde:
if val_loss < THR:
break
Ich habe in der Dokumentation gesehen, dass es die Möglichkeit gibt, einen eigenen Rückruf durchzuführen:http: //keras.io/callbacks Aber nichts gefunden, wie man den Trainingsprozess stoppen kann. Ich brauche einen Rat