浏览代码

Merge branch 'release/v0.1.6'

Jérôme BUISINE 5 年之前
父节点
当前提交
d5de038bdc
共有 1 个文件被更改,包括 10 次插入10 次删除
  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'))