Skip to content

Train, predict, export and reload a tf.estimator for inference

Notifications You must be signed in to change notification settings

guillaumegenthial/tf-estimator-basics

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Tensorflow Estimator Basics

Train, predict, export and reload a tf.estimator for inference on a dummy example.

Read the blog post

Quickstart

make run

Details

  • model.py defines the model_fn
  • train.py trains an Estimator using the model_fn
  • export.py exports the Estimator as a saved_model
  • predict.py reloads an Estimator and uses it for prediction
  • serve.py reloads the inference graph from the saved_model format and uses it for prediction

About

Train, predict, export and reload a tf.estimator for inference

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published