Skip to content

Instantly share code, notes, and snippets.

@vishwanath79
Created June 20, 2020 23:13
Show Gist options
  • Save vishwanath79/8308d2e1ae08b61d8687fd19a1fb994b to your computer and use it in GitHub Desktop.
Save vishwanath79/8308d2e1ae08b61d8687fd19a1fb994b to your computer and use it in GitHub Desktop.
#!/usr/bin/python
# -*- coding: utf-8 -*-
from keras.callbacks import EarlyStopping, ModelCheckpoint
save_early_callback = EarlyStopping(monitor='val_loss', patience=5)
save_best_callback = \
ModelCheckpoint('/content/model-{epoch:02d}-{val_accuracy:.2f}.f5'
, save_best_only=True, save_weights_only=True)
model.fit(
X_train,
y_train,
batch_size=64,
class_weight=class_weight,
validation_split=0.1,
epochs=50,
callbacks=[save_early_callback, save_best_callback],
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment