Update m1_tf_test.py

2022-07-10 15:33:09 +02:00
parent 586d956d28
commit ad47742d32
1 changed files with 2 additions and 1 deletions
@@ -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()