tf.contrib.estimator.build_supervised_input_receiver_fn_from_input_fn
Get a function that returns a SupervisedInputReceiver matching an input_fn.
tf.contrib.estimator.build_supervised_input_receiver_fn_from_input_fn( input_fn, **input_fn_args )
Note that this function calls the input_fn in a local graph in order to extract features and labels. Placeholders are then created from those features and labels in the default graph.
Args | |
---|---|
input_fn | An Estimator input_fn, which is a function that returns one of:
|
**input_fn_args | set of kwargs to be passed to the input_fn. Note that these will not be checked or validated here, and any errors raised by the input_fn will be thrown to the top. |
Returns | |
---|---|
A function taking no arguments that, when called, returns a SupervisedInputReceiver. This function can be passed in as part of the input_receiver_map when exporting SavedModels from Estimator with multiple modes. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/estimator/build_supervised_input_receiver_fn_from_input_fn