Update m1_tf_test.py

2022-07-10 16:39:42 +02:00
parent d8163901e9
commit 6729676855
1 changed files with 4 additions and 3 deletions
@@ -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',