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

Question

我正在尝试将 Tensorflow 数据集的快速绘制数据集加载到 Google Colab。 但由于它太大（~30GB），我只想下载或加载其中的一部分。 我可以在 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%]')