image_output_1.png
import tensorflow as tf

tf.random.set_seed(42)

model = tf.keras.Sequential([tf.keras.layers.Dense(units=784, activation='relu', input_shape=(784,)), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(units=392, activation='relu'), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(units=151, activation='relu'), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(units=50, activation='relu'), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(units=10, activation='softmax')])