Skip to content

Instantly share code, notes, and snippets.

@eisenjulian
Last active March 4, 2018 23:37
Show Gist options
  • Save eisenjulian/a9bec0c011e3be5fb9695d36e5c1c539 to your computer and use it in GitHub Desktop.
Save eisenjulian/a9bec0c011e3be5fb9695d36e5c1c539 to your computer and use it in GitHub Desktop.
initializer = tf.random_uniform([vocab_size, embedding_size], -1.0, 1.0))
params = {'embedding_initializer': initializer}
cnn_classifier = tf.estimator.Estimator(model_fn=model_fn,
model_dir=os.path.join(model_dir, 'cnn'),
params=params)
train_and_evaluate(cnn_classifier)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment