Skip to content

Instantly share code, notes, and snippets.

@swap2ag

swap2ag/tf_model.py

Created June 3, 2020 15:07
Show Gist options
  • Save swap2ag/7ecec338a5a2574f7d60d53708da89be to your computer and use it in GitHub Desktop.
Save swap2ag/7ecec338a5a2574f7d60d53708da89be to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tf_model.py
model = tf.keras.models.Sequential([tf.keras.layers.Dense(units=26, input_shape=(X_train.shape[1],), activation=tf.keras.activations.relu),
tf.keras.layers.Dense(units=26, activation=tf.keras.activations.sigmoid)
])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment