Skip to content

Instantly share code, notes, and snippets.

@jeetaf
Created July 13, 2021 17:18
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jeetaf/6d033fd530a85da78432517bfec6324f to your computer and use it in GitHub Desktop.
Save jeetaf/6d033fd530a85da78432517bfec6324f to your computer and use it in GitHub Desktop.
model = tf.keras.models.Sequential([
tf.keras.layers.Dense(256, input_shape=(9,), activation='relu'),
tf.keras.layers.BatchNormalization(),
tf.keras.layers.Dropout(0.3),
tf.keras.layers.Dense(256, activation='relu'),
tf.keras.layers.BatchNormalization(),
tf.keras.layers.Dropout(0.3),
tf.keras.layers.Dense(512, activation='relu'),
tf.keras.layers.BatchNormalization(),
tf.keras.layers.Dropout(0.5),
tf.keras.layers.Dense(8, activation='softmax')
])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment