Parcourir la source

Model CCN layers Updates

jbuisine il y a 5 ans
Parent
commit
7035cdd867
2 fichiers modifiés avec 0 ajouts et 12 suppressions
  1. 0 4
      classification_cnn_keras.py
  2. 0 8
      classification_cnn_keras_cross_validation.py

+ 0 - 4
classification_cnn_keras.py

@@ -72,10 +72,6 @@ def generate_model():
     model.add(Activation('relu'))
     model.add(MaxPooling2D(pool_size=(2, 2)))
 
-    model.add(Conv2D(10, (2, 2)))
-    model.add(Activation('relu'))
-    model.add(MaxPooling2D(pool_size=(2, 2)))
-
     model.add(Flatten())
 
     model.add(Dense(140))

+ 0 - 8
classification_cnn_keras_cross_validation.py

@@ -70,14 +70,6 @@ def generate_model():
     model.add(Activation('relu'))
     model.add(MaxPooling2D(pool_size=(2, 2)))
 
-    model.add(Conv2D(40, (2, 2)))
-    model.add(Activation('relu'))
-    model.add(MaxPooling2D(pool_size=(2, 2)))
-
-    model.add(Conv2D(20, (2, 2)))
-    model.add(Activation('relu'))
-    model.add(MaxPooling2D(pool_size=(2, 2)))
-
     model.add(Flatten())
 
     model.add(Dense(140))