Skip to main content
Open on GitHub

TensorFlow Datasets

TensorFlow Datasets is a collection of datasets ready to use, with TensorFlow or other Python ML frameworks, such as Jax. All datasets are exposed as tf.data.Datasets, enabling easy-to-use and high-performance input pipelines. To get started see the guide and the list of datasets.

Installation and Setup​

You need to install tensorflow and tensorflow-datasets python packages.

pip install tensorflow
pip install tensorflow-dataset

Document Loader​

See a usage example.

from langchain_community.document_loaders import TensorflowDatasetLoader

Was this page helpful?