Skip to content

Instantly share code, notes, and snippets.

@vishwanath79
Created June 20, 2020 23:12
Show Gist options
  • Save vishwanath79/0ce8fd59ba4ce1ce7f856bed08bc68ca to your computer and use it in GitHub Desktop.
Save vishwanath79/0ce8fd59ba4ce1ce7f856bed08bc68ca to your computer and use it in GitHub Desktop.
#!/usr/bin/python
# -*- coding: utf-8 -*-
from keras.callbacks import EarlyStopping
save_early_callback = EarlyStopping(monitor='val_loss', min_delta=0,
patience=3, verbose=1,
restore_best_weights=True)
model.fit(
X_train,
y_train,
batch_size=64,
class_weight=class_weight,
validation_split=0.1,
epochs=50,
callbacks=[save_early_callback],
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment