Commit fd4cd7b1 authored by Nguyen Ngoc-Son-Dorian's avatar Nguyen Ngoc-Son-Dorian
parents a80abbdf da6cfbda
......@@ -114,7 +114,7 @@ for dataset in WineDatasets:
#MNIST
model = keras.Sequential([
keras.layers.Flatten(input_shape=(Nc,),
keras.layers.Flatten(input_shape=(Nc,)),
keras.layers.Dense(64, 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