Przeglądaj źródła

Merge branch 'release/v0.1.6'

Jérôme BUISINE 4 lat temu
rodzic
commit
d5de038bdc
1 zmienionych plików z 10 dodań i 10 usunięć
  1. 10 10
      models/cnn_models.py

+ 10 - 10
models/cnn_models.py

@@ -29,27 +29,27 @@ def generate_model_2D(_input_shape):
     model.add(Dense(140))
     model.add(Activation('relu'))
     model.add(BatchNormalization())
-    model.add(Dropout(0.4))
+    model.add(Dropout(0.5))
 
     model.add(Dense(120))
     model.add(Activation('relu'))
     model.add(BatchNormalization())
-    model.add(Dropout(0.4))
+    model.add(Dropout(0.5))
 
     model.add(Dense(80))
     model.add(Activation('relu'))
     model.add(BatchNormalization())
-    model.add(Dropout(0.4))
+    model.add(Dropout(0.5))
 
     model.add(Dense(40))
     model.add(Activation('relu'))
     model.add(BatchNormalization())
-    model.add(Dropout(0.4))
+    model.add(Dropout(0.5))
 
     model.add(Dense(20))
     model.add(Activation('relu'))
     model.add(BatchNormalization())
-    model.add(Dropout(0.4))
+    model.add(Dropout(0.5))
 
     model.add(Dense(1))
     model.add(Activation('sigmoid'))
@@ -83,27 +83,27 @@ def generate_model_3D(_input_shape):
     model.add(Dense(140))
     model.add(Activation('relu'))
     model.add(BatchNormalization())
-    model.add(Dropout(0.4))
+    model.add(Dropout(0.5))
 
     model.add(Dense(120))
     model.add(Activation('relu'))
     model.add(BatchNormalization())
-    model.add(Dropout(0.4))
+    model.add(Dropout(0.5))
 
     model.add(Dense(80))
     model.add(Activation('relu'))
     model.add(BatchNormalization())
-    model.add(Dropout(0.4))
+    model.add(Dropout(0.5))
 
     model.add(Dense(40))
     model.add(Activation('relu'))
     model.add(BatchNormalization())
-    model.add(Dropout(0.4))
+    model.add(Dropout(0.5))
 
     model.add(Dense(20))
     model.add(Activation('relu'))
     model.add(BatchNormalization())
-    model.add(Dropout(0.4))
+    model.add(Dropout(0.5))
 
     model.add(Dense(1))
     model.add(Activation('sigmoid'))