Merge pull request #2 from nestoralonso/patch-1

specify the input shape in the first layer

Author: sethjuarez

LearnDigitz/keras_train.py

@@ -50,7 +50,7 @@ def mlp():
 
 def cnn():
   return Sequential([
-    Reshape((28, 28, 1)),
+    Reshape((28, 28, 1), input_shape=(784, )),
     Conv2D(32, [5, 5], padding='same', activation='relu'),
     MaxPooling2D(strides=2),
     Conv2D(64, [5, 5], padding='same', activation='relu'),
@@ -93,4 +93,4 @@ def run(data_dir, model_dir, epochs):
   unique = datetime.now().strftime('%m.%d_%H.%M')
   model_dir = check_dir(os.path.join(output_dir, 'models', 'model_{}'.format(unique)))
   epochs = 10
-  run(data_dir, model_dir, epochs)
+  run(data_dir, model_dir, epochs)