CODEWITHSUNDEEP
pythontensorflowtensorflow-datasets
guidingfox
guidingfox

Reputation: 97

Loading part of a dataset using load() method from Tensorflow datasets

I am trying to load the quick draw dataset from Tensorflow datasets to Google Colab. But since its too large(~30GB) I want to download or load only a part of it. Can I do this using load() method in tensorflow_datasets?

This is the code:

    import tensorflow as tf
    import tensorflow_datasets as tfds

    dataset, metadata = tfds.load(name="quickdraw_bitmap", with_info=True, as_supervised=True)

Thanks in advance.

Upvotes: 1

Views: 914

Answers (1)

Nicolas Gervais
Nicolas Gervais

Reputation: 36584

Yes you can, please read TFDS: Splits and slicing.

dataset, metadata = tfds.load(name="quickdraw_bitmap", 
                              with_info=True, 
                              as_supervised=True,
                              split='train[:10%]')

Upvotes: 1

Related Questions