From ad47742d3269df89a5e2d3d2393a193d2667a5a3 Mon Sep 17 00:00:00 2001 From: Smiril Date: Sun, 10 Jul 2022 15:33:09 +0200 Subject: [PATCH] Update m1_tf_test.py --- m1_tf_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/m1_tf_test.py b/m1_tf_test.py index 434fb95..dd4664d 100644 --- a/m1_tf_test.py +++ b/m1_tf_test.py @@ -17,6 +17,7 @@ model = tf.keras.models.Sequential([ tf.keras.layers.Dense(128, activation='relu'), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(10), + tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(2,activation=tf.nn.softmax) ]) loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True) @@ -26,4 +27,4 @@ model.compile(optimizer='adam', model.fit(x_train, y_train, epochs=100) model.build() model.summary() - +sys.exit()