File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -109,8 +109,12 @@ def __init__(self,
109
109
features: `tfds.features.FeaturesDict`, Information on the feature dict
110
110
of the `tf.data.Dataset()` object from the `builder.as_dataset()`
111
111
method.
112
- supervised_keys: `tuple`, Specifies the input feature and the label for
113
- supervised learning, if applicable for the dataset.
112
+ supervised_keys: `tuple` of size two, optional. The first item is
113
+ the key from `features` corresponding to a single feature
114
+ representing the input, the second item is the key of the
115
+ "feature" containing the label for supervised learning,
116
+ if applicable for the dataset. Used for transforming the dataset in
117
+ `tfds.core.DatasetBuilder.as_dataset()`.
114
118
urls: `list(str)`, optional, the homepage(s) for this dataset.
115
119
citation: `str`, optional, the citation to use for this dataset.
116
120
metadata: `tfds.core.Metadata`, additonal object which will be
You can’t perform that action at this time.
0 commit comments