From 67296768555b33885ad4ee769d2b446bd8191648 Mon Sep 17 00:00:00 2001 From: Smiril Date: Sun, 10 Jul 2022 16:39:42 +0200 Subject: [PATCH] Update m1_tf_test.py --- m1_tf_test.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/m1_tf_test.py b/m1_tf_test.py index 8afb513..c6fd3fa 100644 --- a/m1_tf_test.py +++ b/m1_tf_test.py @@ -14,11 +14,12 @@ mnist = tf.keras.datasets.mnist x_train, x_test = x_train / 255.0, x_test / 255.0 model = tf.keras.models.Sequential([ tf.keras.layers.Flatten(input_shape=(28, 28)), - tf.keras.layers.Dense(128, activation='relu'), + tf.keras.layers.Dense(128,activation=tf.nn.softmax), tf.keras.layers.Dropout(0.2), - tf.keras.layers.Dense(10), + tf.keras.layers.Dense(10,activation=tf.nn.softmax), tf.keras.layers.Dropout(0.2), - tf.keras.layers.Dense(2,activation=tf.nn.softmax) + tf.keras.layers.Dense(2,activation=tf.nn.softmax), + tf.keras.layers.Dropout(0.2) ]) loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True) model.compile(optimizer='adam',