Commit 8a0b5796 authored by Shivani Agrawal's avatar Shivani Agrawal Committed by TensorFlower Gardener
Browse files

[tf.data] Adds an optional label_key argument to... 

[tf.data] Adds an optional label_key argument to `make_batch_featured_dataset()` for extracting the label from the feature dictionaries. If label_key is provided, returned dataset will be a tuple of feature dictionaries and label.

PiperOrigin-RevId: 210766469
parent c69f998b
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment