Sfoglia il codice sorgente

Model CCN layers Updates

jbuisine 6 anni fa
parent
commit
7035cdd867

+ 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))