Module: tfds

View source on GitHub

tensorflow_datasets (tfds) defines a collection of datasets ready-to-use with TensorFlow.

Each dataset is defined as a tfds.core.DatasetBuilder, which encapsulates the logic to download the dataset and construct an input pipeline, as well as contains the dataset documentation (version, splits, number of examples, etc.).

The main library entrypoints are:

• tfds.builder: fetch a tfds.core.DatasetBuilder by name
• tfds.load: convenience method to construct a builder, download the data, and create an input pipeline, returning a tf.data.Dataset.

Documentation:

• These API docs
• Available datasets
• Colab tutorial
• Add a dataset

Modules

beam module: Beam utils.

core module: API to define datasets.

dataset_builders module: Dataset builders API.

decode module: Decoder public API.

deprecated module: Deprecated symbols.

download module: tfds.download.DownloadManager API.

features module: API defining dataset features (image, text, scalar,...).