#!/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],
    )