Hugging Face's logo

Spaces:
crazyscientist1
/
microScan
Sleeping

App Files Community
crazyscientist1 commited on
Commit
866d13c
•
1 Parent(s): 963d4f4

Update appTest.py

Browse files
Files changed (1) hide show
  1. 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.2)(top_model)
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)