tf-estimator-basics
tf-estimator-basics copied to clipboard
Train, predict, export and reload a tf.estimator for inference
Tensorflow Estimator Basics
Train, predict, export and reload a tf.estimator for inference on a dummy example.
Quickstart
make run
Details
model.pydefines themodel_fntrain.pytrains an Estimator using themodel_fnexport.pyexports the Estimator as asaved_modelpredict.pyreloads an Estimator and uses it for predictionserve.pyreloads the inference graph from thesaved_modelformat and uses it for prediction