Parcourir la source

Merge branch 'release/v0.1.6'

Jérôme BUISINE il y a 4 ans
Parent
commit
d5de038bdc
1 fichiers modifiés avec 10 ajouts et 10 suppressions
  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'))