Commit 69b18b93 authored by Ngocson's avatar Ngocson

correction

parent 1a319259
......@@ -73,7 +73,7 @@ for dataset in MNISTdatasets:
#MNIST
model = keras.Sequential([
keras.layers.Flatten(input_shape=(Nc)),
keras.layers.Flatten(input_shape=(Nc,)),
keras.layers.Dense(128, activation=tf.nn.tanh),
keras.layers.Dense(64, activation=tf.nn.tanh),
keras.layers.Dense(32, activation=tf.nn.tanh),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment