TensorFlow从numpy数组创建数据集

Don*_*beo 14 python machine-learning mnist tensorflow

TensorFlow构建了一种存储数据的好方法.例如,这用于在示例中存储MNIST数据:

>>> mnist
<tensorflow.examples.tutorials.mnist.input_data.read_data_sets.<locals>.DataSets object at 0x10f930630>
Run Code Online (Sandbox Code Playgroud)

假设有一个输入和输出numpy数组.

>>> x = np.random.normal(0,1, (100, 10))
>>> y = np.random.randint(0, 2, 100)
Run Code Online (Sandbox Code Playgroud)

如何在tf数据集中转换它们?

我想使用像这样的功能 next_batch

Ian*_*low 8

Dataset对象只是MNIST教程的一部分,而不是主要的TensorFlow库.

您可以在此处查看其定义位置:

GitHub链接

构造函数接受一个图像和标签参数,所以大概你可以在那里传递你自己的值.