Skip to content

Instantly share code, notes, and snippets.

@lakshmanok

lakshmanok/model_train_eval.py

Created August 13, 2018 16:16
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save lakshmanok/6add2c1503f416c6d502d0fef1ac5a53 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save lakshmanok/6add2c1503f416c6d502d0fef1ac5a53 to your computer and use it in GitHub Desktop.
Download ZIP
How to serve an embedding trained with Estimators
Raw
model_train_eval.py
def train_and_evaluate(output_dir, nsteps):
estimator = tf.estimator.Estimator(
model_fn = model_fn,
model_dir = output_dir)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment