Skip to content

Instantly share code, notes, and snippets.

@rdisipio
Created November 6, 2022 13:40
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 rdisipio/d04469b96080f9307785d8c09ec9da05 to your computer and use it in GitHub Desktop.
Save rdisipio/d04469b96080f9307785d8c09ec9da05 to your computer and use it in GitHub Desktop.
model1 = tf.keras.Sequential([
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dense(2, activation='softmax') # binary classification
])
loss = tf.keras.losses.BinaryCrossentropy(from_logits=False)
metrics = [tf.keras.metrics.BinaryAccuracy()]
model1.compile(optimizer='adam',
loss=loss,
metrics=metrics)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment