【发布时间】:2016-05-01 02:08:13
【问题描述】:
在训练我的 MNIST 分类网络后,我想对测试数据进行“预测”,但在测试输入的形状方面出现以下错误
testimages = np.array(test)
print(testimages.shape)
> (28000, 784)
feed_dict = {x: [testimages]}
classification = sess.run(y, feed_dict)
ValueError: 无法为具有形状 (Dimension(None), Dimension(784)) 的张量 u'Placeholder_2:0' 提供形状 (1, 28000, 784) 的值
那么形状怎么可能是 (28000, 784)(它应该是),但是当输入到经过训练的网络时,它会显示为 (1, 28000, 784)?
顺便说一句,为了训练,我通过
包含了训练数据trainlabels = np.array(train["label"])
trainimages = np.array(train.iloc[:, 1:])
因为训练数据的第一列说明了标签。我正在使用 Pandas 进行导入。
【问题讨论】:
标签: python tensorflow mnist