tf.estimator.experimental.build_raw_supervised_input_receiver_fn
Build a supervised_input_receiver_fn for raw features and labels.
tf.estimator.experimental.build_raw_supervised_input_receiver_fn( features, labels, default_batch_size=None )
This function wraps tensor placeholders in a supervised_receiver_fn with the expectation that the features and labels appear precisely as the model_fn expects them. Features and labels can therefore be dicts of tensors, or raw tensors.
Args | |
---|---|
features | a dict of string to Tensor or Tensor . |
labels | a dict of string to Tensor or Tensor . |
default_batch_size | the number of query examples expected per batch. Leave unset for variable batch size (recommended). |
Returns | |
---|---|
A supervised_input_receiver_fn. |
Raises | |
---|---|
ValueError | if features and labels have overlapping keys. |
© 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/r2.4/api_docs/python/tf/estimator/experimental/build_raw_supervised_input_receiver_fn