tfds.decode.PartialDecoding

Allow to decode a subset of features.

tfds.decode.PartialDecoding(
    features: tfds.typing.FeatureSpecs,
    decoders: Optional[utils.TreeDict[base.Decoder]] = None
)

See guide:

https://www.tensorflow.org/datasets/decode#only_decode_a_sub-set_of_the_features

features A nested dict of tfds.features.FeatureConnector
decoders Optional additional decoders to apply (e.g. tfds.decode.SkipDecoding())

Methods

extract_features

View source

extract_features(
    features: tfds.features.FeatureConnector
) -> tfds.features.FeatureConnector

Returns the tfds.features.FeaturesDict.

Extract the subset of features

Args
features Features on which extract the sub-set

Returns
features_subset A subset of the features