tensorflow.examples.tutorials.mnist 中的数据是如何格式化的?

数据挖掘 机器学习 神经网络 深度学习 张量流 极简主义
2022-01-27 23:03:16

我正在分析使用 Tensorflow 实现受限玻尔兹曼机 (RBM) 的示例代码。输入数据似乎是良好的旧 MNIST,但显然它现在Tensorflow本身可用。

因此,我不想在 MNIST 上运行这个示例代码,而是想在我自己的数据上运行它。但问题是我不知道如何

tensorflow.examples.tutorials.mnist

已格式化。它像一个CSV文件吗?如果我想提供来自 Wizemann 的马图像的自己的数据,我应该如何格式化与 MNIST 的格式化方式兼容的数据?

我对这些东西很陌生。所以,如果我弄错了或者我的问题很荒谬,请赐教。

1个回答

加载数据后根据官方文档,

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)

您可以使用以下行,

mnist.train.images.shape

查看图像的形状,如您所见,它是一个 rank-2 矩阵,这意味着图像被展开,其中一个维度指定示例数,另一个指定像素数 __features。