From f85acaa73e994abace063094e569e2cf4bb57835 Mon Sep 17 00:00:00 2001 From: Kamila Bobkowska Date: Mon, 11 May 2020 09:26:17 +0000 Subject: [PATCH] changing a mistake --- checkingTrash.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/checkingTrash.py b/checkingTrash.py index 9aa5087..b87b1fb 100644 --- a/checkingTrash.py +++ b/checkingTrash.py @@ -42,7 +42,7 @@ classifier.add(MaxPooling2D(pool_size=(2, 2))) classifier.add(Flatten()) classifier.add(Dense(activation = "relu", units = 64 )) classifier.add(Dense(activation = "softmax", units = 5)) -classifier.compile(optimizer = "adam", loss = "binary_crossentropy", metrics = ["accuracy"]) +classifier.compile(optimizer = "adam", loss = "categorical_crossentropy", metrics = ["accuracy"]) @@ -80,14 +80,14 @@ test_generator = test_datagen.flow_from_directory( #Teaching the classifier '''classifier.fit_generator( train_generator, steps_per_epoch = 165, epochs = 32, validation_data = test_generator ) -classifier.save_weights('model_ver_5.h5')''' +classifier.save_weights('model_ver_6.h5')''' labels = (train_generator.class_indices) labels = dict((value,key) for key,value in labels.items()) -classifier.load_weights("model_ver_5.h5") +classifier.load_weights("model_ver_6.h5")