Skip to content

Instantly share code, notes, and snippets.

@ispamm
Created March 5, 2018 17:09
Show Gist options
  • Save ispamm/15cd3a5b1f448db00bb87a01705ca0d8 to your computer and use it in GitHub Desktop.
Save ispamm/15cd3a5b1f448db00bb87a01705ca0d8 to your computer and use it in GitHub Desktop.
hid = tf.layers.Dense(units=10, activation=tf.nn.relu)
drop = tf.layers.Dropout()
out = tf.layers.Dense(units=3, activation=None)
def nn_model(x, training=False):
return out(drop(hid(x), training=training))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment