From 4af43d255d072920ed613a9a29bbc7127c42a8c5 Mon Sep 17 00:00:00 2001 From: Kamran Kowsari Date: Mon, 29 Jun 2020 10:17:57 -0700 Subject: [PATCH] Update RMDL_Image.py --- RMDL/RMDL_Image.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/RMDL/RMDL_Image.py b/RMDL/RMDL_Image.py index 762649f..7ab4018 100644 --- a/RMDL/RMDL_Image.py +++ b/RMDL/RMDL_Image.py @@ -144,7 +144,7 @@ def Image_Classification(x_train, y_train, x_test, y_test, shape, batch_size=128 filepath = "weights\weights_DNN_" + str(i) + ".hdf5" checkpoint = ModelCheckpoint(filepath, - monitor='val_acc', + monitor='val_accuracy', verbose=1, save_best_only=True, mode='max') @@ -197,7 +197,7 @@ def Image_Classification(x_train, y_train, x_test, y_test, shape, batch_size=128 filepath = "weights\weights_RNN_" + str(i) + ".hdf5" checkpoint = ModelCheckpoint(filepath, - monitor='val_acc', + monitor='val_accuracy', verbose=1, save_best_only=True, mode='max') @@ -247,7 +247,10 @@ def Image_Classification(x_train, y_train, x_test, y_test, shape, batch_size=128 dropout) filepath = "weights\weights_CNN_" + str(i) + ".hdf5" - checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=True, + checkpoint = ModelCheckpoint(filepath, + monitor='val_accuracy', + verbose=1, + save_best_only=True, mode='max') callbacks_list = [checkpoint]