文件名称:TensorFlow Datasets - 准备与TensorFlow一起使用的数据集的集合-python
文件大小:182.6MB
文件格式:ZIP
更新时间:2024-07-08 12:14:07
机器学习
TensorFlow Datasets - 准备与TensorFlow一起使用的数据集的集合 TensorFlow 数据集 注意:tensorflow_datasets 尚不可用,正在积极开发中。 API 不稳定。 TensorFlow Datasets 提供了许多公共数据集作为 tf.data.Datasets。 安装:pip install tensorflow-datasets # 还需要安装tensorflow或tensorflow-gpu 用法:import tensorflow_datasets as datasets #构造一个tf.data.Dataset dataset = datasets.load("mnist", split="train", data_dir= "~/tfdata", download=True) # 构建你的输入管道 dataset = dataset.shuffle(1000).batch(128).prefetch(1) features = dataset.make_oneshot_iterator().get_next() im