tf.contrib.predictor.from_saved_model
Constructs a Predictor
from a SavedModel
on disk.
tf.contrib.predictor.from_saved_model(
export_dir, signature_def_key=None, signature_def=None, input_names=None,
output_names=None, tags=None, graph=None, config=None
)
Args |
export_dir | a path to a directory containing a SavedModel . |
signature_def_key | Optional string specifying the signature to use. If None , then DEFAULT_SERVING_SIGNATURE_DEF_KEY is used. Only one of signature_def_key and signature_def |
signature_def | A SignatureDef proto specifying the inputs and outputs for prediction. Only one of signature_def_key and signature_def should be specified. input_names: A dictionary mapping strings to Tensor s in the SavedModel that represent the input. The keys can be any string of the user's choosing. output_names: A dictionary mapping strings to Tensor s in the SavedModel that represent the output. The keys can be any string of the user's choosing. |
tags | Optional. Tags that will be used to retrieve the correct SignatureDef . Defaults to DEFAULT_TAGS . |
graph | Optional. The Tensorflow graph in which prediction should be done. |
config | ConfigProto proto used to configure the session. |
Returns |
An initialized Predictor . |
Raises |
ValueError | More than one of signature_def_key and signature_def is specified. |