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()