Skip to content

Instantly share code, notes, and snippets.

@eisenjulian
Last active March 4, 2018 23:37
Show Gist options
  • Save eisenjulian/38a4db49755a6e790763768a75556379 to your computer and use it in GitHub Desktop.
Save eisenjulian/38a4db49755a6e790763768a75556379 to your computer and use it in GitHub Desktop.
training = (mode == tf.estimator.ModeKeys.TRAIN)
dropout_emb = tf.layers.dropout(inputs=input_layer,
rate=0.2,
training=training)
conv = tf.layers.conv1d(
inputs=dropout_emb,
filters=32,
kernel_size=3,
padding="same",
activation=tf.nn.relu)
pool = tf.reduce_max(input_tensor=conv, axis=1)
hidden = tf.layers.dense(inputs=pool, units=250, activation=tf.nn.relu)
dropout = tf.layers.dropout(inputs=hidden, rate=0.2, training=training)
logits = tf.layers.dense(inputs=dropout_hidden, units=1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment