From 69022561f0a2bc73320ca5f8069bea91ae62151a Mon Sep 17 00:00:00 2001 From: marcsello Date: Mon, 14 Sep 2020 15:16:05 +0200 Subject: [PATCH] Fixed RMSProp --- cnn_classification_service/cnn_classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cnn_classification_service/cnn_classifier.py b/cnn_classification_service/cnn_classifier.py index ec61935..df8570b 100644 --- a/cnn_classification_service/cnn_classifier.py +++ b/cnn_classification_service/cnn_classifier.py @@ -22,7 +22,7 @@ class Classifier(object): self.loaded_model.load_weights(weights_filename) self.datagen = ImageDataGenerator(rescale=1. / 255., validation_split=0.25) - self.loaded_model.compile(optimizers.rmsprop(lr=0.0005, decay=1e-6), loss="categorical_crossentropy", + self.loaded_model.compile(optimizers.RMSprop(lr=0.0005, decay=1e-6), loss="categorical_crossentropy", metrics=["accuracy"]) self.loaded_model.summary()