Skip to content

Instantly share code, notes, and snippets.

@sergei-mironov
Last active November 28, 2019 10:21
Show Gist options
  • Select an option

  • Save sergei-mironov/af19082d6fdfd95f86e1b342d3c2d997 to your computer and use it in GitHub Desktop.

Select an option

Save sergei-mironov/af19082d6fdfd95f86e1b342d3c2d997 to your computer and use it in GitHub Desktop.
demo_func.py
inputs = keras.Input(shape=(784,), name='digits')
x = layers.Dense(64, activation='relu', name='dense_1')(inputs)
x = layers.Dense(64, activation='relu', name='dense_2')(x)
outputs = layers.Dense(10, activation='softmax', name='predictions')(x)
model = keras.Model(inputs=inputs, outputs=outputs)
model.fit(...)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment