Skip to content

Instantly share code, notes, and snippets.

@nbortolotti

nbortolotti/keras_example_irismodelbuild.py

Created January 7, 2019 12:53
Show Gist options
  • Save nbortolotti/300077fb770a2d93de22c8ff2ac8d1a4 to your computer and use it in GitHub Desktop.
Save nbortolotti/300077fb770a2d93de22c8ff2ac8d1a4 to your computer and use it in GitHub Desktop.
Download ZIP
Keras Model definition
Raw
keras_example_irismodelbuild.py
model = tf.keras.Sequential([
tf.keras.layers.Dense(16, input_dim=4),
tf.keras.layers.Dense(3, activation=tf.nn.softmax),
])
model.compile(loss='categorical_crossentropy',
optimizer='sgd',
metrics=['accuracy'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment