diff options
author | Kiril Gorovoy <kgorovoy@google.com> | 2017-03-10 10:09:30 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-10 10:29:07 -0800 |
commit | 0da4acdf225753becca25b9dc9127cca7b2bc465 (patch) | |
tree | 5a41162fe2babe8089b22472562f8362c6060a56 /tensorflow/examples/saved_model | |
parent | b92f7aa6fea523fb3521da331dec43aba521ec81 (diff) |
Add a classification signature to the test saved_model_half_plus_two_model.
Test using this classification signature using the Predict API.
Change: 149769547
Diffstat (limited to 'tensorflow/examples/saved_model')
-rw-r--r-- | tensorflow/examples/saved_model/saved_model_half_plus_two.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/examples/saved_model/saved_model_half_plus_two.py b/tensorflow/examples/saved_model/saved_model_half_plus_two.py index 52c95190bc..75cfcdb19e 100644 --- a/tensorflow/examples/saved_model/saved_model_half_plus_two.py +++ b/tensorflow/examples/saved_model/saved_model_half_plus_two.py @@ -181,6 +181,8 @@ def _generate_saved_model_for_half_plus_two(export_dir, as_text=False): _build_regression_signature(x2, y3), "classify_x_to_y": _build_classification_signature(serialized_tf_example, y), + "classify_x2_to_y3": + _build_classification_signature(x2, y3), tf.saved_model.signature_constants. DEFAULT_SERVING_SIGNATURE_DEF_KEY: predict_signature_def |