Convert the dataset into a pandas dataframe.

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

ds The 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

dataframe The pandas.DataFrame object