birbnetes/cnn-classification-service
birbnetes/cnn-classification-service
4
0
Fork 0
Code Pull Requests Releases Wiki Activity

Fixed RMSProp

Browse Source
This commit is contained in:
Pünkösd Marcell 2020-09-14 15:16:05 +02:00
parent 2cae318d1d
commit 69022561f0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cnn_classification_service/cnn_classifier.py
View File

@ -22,7 +22,7 @@ class Classifier(object):
self.loaded_model.load_weights(weights_filename) self.loaded_model.load_weights(weights_filename)
self.datagen = ImageDataGenerator(rescale=1. / 255., validation_split=0.25) 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"]) metrics=["accuracy"])
self.loaded_model.summary() self.loaded_model.summary()