reordered imports for pep8 compatibility
@@ -7,12 +7,13 @@ from distutils.version import LooseVersion
import keras
from keras.models import Sequential
from keras.layers import Dense, Dropout, Activation, Flatten
+import matplotlib.pyplot as plt
+import os
+
if LooseVersion(keras.__version__[0]) >= LooseVersion('2.0.0'):
from keras.layers import Conv2D
else:
from keras.layers import Convolution2D
-import matplotlib.pyplot as plt
-import os
class _ArgsWrapper(object):
@@ -131,7 +132,6 @@ def conv_wrap(filters, kernel, strides, padding):
border_mode=padding)
-
def cnn_model(logits=False, input_ph=None, img_rows=28, img_cols=28,
channels=1, nb_filters=64, nb_classes=10):
"""
