s444517/AI2020_Project
1
2
Fork 0
Code Issues Pull Requests Releases Wiki Activity

changing a mistake

Browse Source
This commit is contained in:
Kamila Bobkowska 2020-05-11 09:25:37 +00:00
parent aa0b9411c6
commit c7ecb5835f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models/Garbagetruck.py
View File

@ -158,7 +158,7 @@ class GarbageTruck():
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"])
train_datagen = ImageDataGenerator(
@ -196,7 +196,7 @@ class GarbageTruck():
test_x, test_y = test_generator.__getitem__(1)
whatarewe = []
classifier.load_weights("model_ver_5.h5")
classifier.load_weights("model_ver_6.h5")
''' #optional for displaying the trash and what the CNN decided it is
z=1
'''