changing a mistake
This commit is contained in:
parent
aa0b9411c6
commit
c7ecb5835f
@ -158,7 +158,7 @@ class GarbageTruck():
|
|||||||
classifier.add(Flatten())
|
classifier.add(Flatten())
|
||||||
classifier.add(Dense(activation = "relu", units = 64 ))
|
classifier.add(Dense(activation = "relu", units = 64 ))
|
||||||
classifier.add(Dense(activation = "softmax", units = 5))
|
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(
|
train_datagen = ImageDataGenerator(
|
||||||
@ -196,7 +196,7 @@ class GarbageTruck():
|
|||||||
test_x, test_y = test_generator.__getitem__(1)
|
test_x, test_y = test_generator.__getitem__(1)
|
||||||
|
|
||||||
whatarewe = []
|
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
|
''' #optional for displaying the trash and what the CNN decided it is
|
||||||
z=1
|
z=1
|
||||||
'''
|
'''
|
||||||
|
Loading…
Reference in New Issue
Block a user