使用load()方法从Tensorflow数据集中加载数据集的一部分

Question

我正在尝试将quick draw dataset从Tensorflow数据集加载到Google Colab。但由于它太大(约30 it )，我只想下载或加载它的一部分。我能在tensorflow_datasets中使用load()方法来做这件事吗？

代码如下：

    import tensorflow as tf
    import tensorflow_datasets as tfds

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

提前谢谢。

Answer 1

是的，你可以，请阅读TFDS: Splits and slicing。

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