Browse Source

Fix cross validation issue

jbuisine 3 years ago
parent
commit
fe5a286edf
1 changed files with 1 additions and 1 deletions
  1. 1 1
      classification_cnn_keras_cross_validation.py

+ 1 - 1
classification_cnn_keras_cross_validation.py

@@ -29,7 +29,7 @@ import json
 from keras.preprocessing.image import ImageDataGenerator
 from keras.models import Sequential
 from keras.layers import Conv2D, MaxPooling2D, AveragePooling2D
-from keras.layers import Activation, Dropout, Flatten, Dense
+from keras.layers import Activation, Dropout, Flatten, Dense, BatchNormalization
 from keras import backend as K
 from keras.utils import plot_model