tfds.as_dataframe

Convert the dataset into a pandas dataframe.

tfds.as_dataframe(
    ds: tf.data.Dataset, ds_info: Optional[dataset_info.DatasetInfo] = None
) -> pd.DataFrame

Used in the notebooks

Used in the tutorials
Custom training: walkthrough TensorFlow Datasets Linear Mixed-Effect Regression in {TF Probability, R, Stan} Fitting Generalized Linear Mixed-effects Models Using Variational Inference Multilevel Modeling Primer in TensorFlow Probability

Warning: The dataframe will be loaded entirely in memory, you may want to call tfds.as_dataframe on a subset of the data instead:

df = tfds.as_dataframe(ds.take(10), ds_info)

Args
ds	tf.data.Dataset. The tf.data.Dataset object to convert to panda dataframe. Examples should not be batched. The full dataset will be loaded.
ds_info	Dataset info object. If given, helps improving the formatting. Available either through tfds.load('mnist', with_info=True) or tfds.builder('mnist').info

Returns
dataframe	The pandas.DataFrame object