Parcourir la source

Fix cross validation issue

jbuisine il y a 5 ans
Parent
commit
fe5a286edf
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  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