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