Spaces:
Sleeping
Sleeping
Commit
•
866d13c
1
Parent(s):
963d4f4
Update appTest.py
Browse files- appTest.py +1 -2
appTest.py
CHANGED
@@ -64,9 +64,8 @@ def load_model():
|
|
64 |
top_model = Dense(4096, activation='relu')(top_model)
|
65 |
top_model = Dense(1048, activation='relu')(top_model)
|
66 |
top_model = Dense(256, activation='relu')(top_model)
|
67 |
-
top_model = Dense(128, activation='relu')(top_model)
|
68 |
top_model = Dense(64, activation='relu')(top_model)
|
69 |
-
top_model = Dropout(0.
|
70 |
output_layer = Dense(5, activation='softmax')(top_model)
|
71 |
|
72 |
CNN = Model(inputs=conv_base.input, outputs=output_layer)
|
|
|
64 |
top_model = Dense(4096, activation='relu')(top_model)
|
65 |
top_model = Dense(1048, activation='relu')(top_model)
|
66 |
top_model = Dense(256, activation='relu')(top_model)
|
|
|
67 |
top_model = Dense(64, activation='relu')(top_model)
|
68 |
+
top_model = Dropout(0.1)(top_model)
|
69 |
output_layer = Dense(5, activation='softmax')(top_model)
|
70 |
|
71 |
CNN = Model(inputs=conv_base.input, outputs=output_layer)
|