TensorFlow Datasets - 准备与TensorFlow一起使用的数据集的集合-python

时间:2021-06-18 18:27:27
【文件属性】:
文件名称:TensorFlow Datasets - 准备与TensorFlow一起使用的数据集的集合-python
文件大小:182.6MB
文件格式:ZIP
更新时间:2021-06-18 18:27:27
机器学习 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

网友评论